fix for test

mars-project · Sep 6, 2021 · b5769cb · b5769cb
1 parent ad14c62
commit b5769cb
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 18 deletions.
diff --git a/mars/deploy/oscar/session.py b/mars/deploy/oscar/session.py
@@ -921,7 +921,7 @@ async def destroy(self):
         await self._session_api.delete_session(self._session_id)
         if self._asyncio_task_timeout_detector_task:  # pragma: no cover
             self._asyncio_task_timeout_detector_task.cancel()
-    
+
     async def create_remote_object(self,
                                    session_id: str,
                                    name: str,
@@ -1123,7 +1123,7 @@ async def create_mutable_tensor(self,
                                     chunksize,
                                     name: str = None):
         pass  # pragma: no cover
-    
+
     @implements(AbstractAsyncSession.get_web_endpoint)
     @_delegate_to_isolated_session
     async def get_web_endpoint(self) -> Optional[str]:

diff --git a/mars/services/mutable/supervisor/service.py b/mars/services/mutable/supervisor/service.py
@@ -5,6 +5,7 @@
 from .... import oscar as mo
 from ..worker.service import MutableTensorChunkActor
 
+
 class MutableTensorActor(mo.Actor):
     def __init__(self, shape: tuple, dtype: str, chunksize, name: str = None):
         self._shape = shape
@@ -61,7 +62,7 @@ async def write(self, index,value):
 
     async def read(self, index):
         result = split_indexes_into_chunks(self._nsplits,index)
-        ans=[]
+        ans_list = []
         for idx,v in result[0].items():
             if len(v[0] > 0):
                 target_index = 0
@@ -75,17 +76,5 @@ async def read(self, index):
                 for nidx in v:
                     val = await chunk_actor.read(idx,nidx)
                     print(idx,nidx,val)
-                    ans.append(val)
-        return ans
-
-    @property
-    def shape(self):
-        return self._shape
-
-    @property
-    def dtype(self):
-        return self._dtype
-
-    @property
-    def name(self):
-        return self._name
+                    ans_list.append(val)
+        return ans_list
diff --git a/mars/services/session/api/core.py b/mars/services/session/api/core.py
@@ -101,9 +101,11 @@ async def fetch_tileable_op_logs(self,
 
     @abstractmethod
     async def create_mutable_tensor(self,
+                                    session_id: str,
                                     shape: tuple,
+                                    dtype: str,
                                     chunksize,
-                                    dtype: str):
+                                    name: str = None):
         """
         Create a mutable tensor.
         """