Merge pull request intel-analytics#4710 from dding3/fixpythontfparkII

add resources for python tfpark
Le-Zheng · Sep 10, 2021 · e239d75 · e239d75
2 parents 32681f6 + 83c343b
commit e239d75
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/python/orca/src/test/bigdl/orca/resources/tfrecord/mnist_test.tfrecord b/python/orca/src/test/bigdl/orca/resources/tfrecord/mnist_test.tfrecord
diff --git a/python/orca/src/test/bigdl/orca/resources/tfrecord/mnist_train.tfrecord b/python/orca/src/test/bigdl/orca/resources/tfrecord/mnist_train.tfrecord
diff --git a/python/orca/src/test/bigdl/orca/tfpark/test_tf_dataset.py b/python/orca/src/test/bigdl/orca/tfpark/test_tf_dataset.py
@@ -60,7 +60,7 @@ def parse_fn(example):
 class TestTFDataset(ZooTestCase):
 
     def get_raw_image_set(self, with_label):
-        resource_path = os.path.join(os.path.split(__file__)[0], "../../resources")
+        resource_path = os.path.join(os.path.split(__file__)[0], "../resources")
         if with_label:
             image_folder = os.path.join(resource_path, "cat_dog")
         else:

diff --git a/python/orca/src/test/bigdl/orca/tfpark/test_tfpark_estimator.py b/python/orca/src/test/bigdl/orca/tfpark/test_tfpark_estimator.py
@@ -26,7 +26,7 @@
 
 class TestTFParkEstimator(ZooTestCase):
     def get_raw_image_set(self, with_label):
-        resource_path = os.path.join(os.path.split(__file__)[0], "../../resources")
+        resource_path = os.path.join(os.path.split(__file__)[0], "../resources")
         if with_label:
             image_folder = os.path.join(resource_path, "cat_dog")
         else: