FIX-modin-project#2844: Correct object reference type to ObjectIDType.

Signed-off-by: Rehan S. Durrani <rdurrani@berkeley.edu>
devin-petersohn · Mar 12, 2021 · 2e33d9c · 2e33d9c
1 parent 470880d
commit 2e33d9c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modin/engines/ray/pandas_on_ray/frame/partition.py b/modin/engines/ray/pandas_on_ray/frame/partition.py
@@ -213,7 +213,7 @@ def ip(self):
  self.drain_call_queue()
  else:
  self._ip_cache = self.apply(lambda df: df)._ip_cache
- if isinstance(self._ip_cache, (ray.ObjectID, ClientObjectRef)):
+ if isinstance(self._ip_cache, ObjectIDType):
  try:
  self._ip_cache = ray.get(self._ip_cache)
  except RayTaskError as e: