remove ray file level imports from shard.py (intel-analytics#3030)

yangw1234 · Sep 26, 2021 · 80a3935 · 80a3935
1 parent f14efe5
commit 80a3935
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/orca/src/bigdl/orca/data/shard.py b/python/orca/src/bigdl/orca/data/shard.py
@@ -15,7 +15,6 @@
 #
 from py4j.protocol import Py4JError
 
-from zoo.orca.data.ray_rdd import RayRdd
 from zoo.orca.data.utils import *
 from zoo.orca import OrcaContext
 from zoo.common.nncontext import init_nncontext
@@ -160,6 +159,7 @@ def to_spark_xshards(self):
 
     @staticmethod
     def from_spark_xshards(spark_xshards):
+        from zoo.orca.data.ray_rdd import RayRdd
         ray_rdd = RayRdd.from_spark_rdd(spark_xshards.rdd)
         return RayXShards(ray_rdd)