FIX-#2688: Address comments

Signed-off-by: Devin Petersohn <devin.petersohn@gmail.com>
modin-project · Feb 5, 2021 · 109c48f · 109c48f
1 parent 877f5ff
commit 109c48f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/modin/experimental/engines/omnisci_on_ray/frame/axis_partition.py b/modin/experimental/engines/omnisci_on_ray/frame/axis_partition.py
@@ -25,7 +25,7 @@ def __init__(self, list_of_blocks):
         self.list_of_blocks = [obj.oid for obj in list_of_blocks]
 
     partition_type = OmnisciOnRayFramePartition
-    instance_type = ray.ObjectID
+    instance_type = ray.ObjectRef
 
 
 class OmnisciOnRayFrameColumnPartition(OmnisciOnRayFrameAxisPartition):

diff --git a/modin/experimental/engines/omnisci_on_ray/frame/partition.py b/modin/experimental/engines/omnisci_on_ray/frame/partition.py
@@ -23,7 +23,7 @@ class OmnisciOnRayFramePartition(BaseFramePartition):
     def __init__(
         self, object_id=None, frame_id=None, arrow_table=None, length=None, width=None
     ):
-        assert type(object_id) is ray.ObjectID
+        assert type(object_id) is ray.ObjectRef
 
         self.oid = object_id
         self.frame_id = frame_id