wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-17 11:25:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

rename test_path -> worker_path (#128)

Browse Source
This commit is contained in:
Robert Nishihara
2016-06-19 22:01:31 -07:00
committed by Philipp Moritz
parent aaa0a514fa
commit ac65f1031e
7 changed files with 29 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/datasets_test.py
+3 -4
View File
@@ -8,11 +8,10 @@ import ray.datasets.imagenet as imagenet
class ImageNetTest(unittest.TestCase):
def testImageNetLoading(self):
test_dir = os.path.dirname(os.path.abspath(__file__))
test_path = os.path.join(test_dir, "test_worker.py")
services.start_singlenode_cluster(return_drivers=False, num_workers_per_objstore=5, worker_path=test_path)
worker_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "test_worker.py")
services.start_singlenode_cluster(return_drivers=False, num_workers_per_objstore=5, worker_path=worker_path)
chunk_name = os.path.join(test_dir, "..", "data", "mini.tar")
chunk_name = os.path.join(os.path.dirname(os.path.abspath(__file__)), "../data/mini.tar")
tar = tarfile.open(chunk_name, mode= "r")
chunk = imagenet.load_chunk(tar, size=(256, 256))
self.assertEqual(chunk.shape, (2, 256, 256, 3))