From 80a3935450614c2cc98c716d0c6bcd094174cb73 Mon Sep 17 00:00:00 2001
From: Yang Wang <yang3.wang@intel.com>
Date: Tue, 10 Nov 2020 13:09:14 +0800
Subject: [PATCH] remove ray file level imports from shard.py (#3030)

---
 python/orca/src/bigdl/orca/data/shard.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/orca/src/bigdl/orca/data/shard.py b/python/orca/src/bigdl/orca/data/shard.py
index ac2efd148a0..805b3396256 100644
--- 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)