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

clean up imports (#230)

Browse Source
This commit is contained in:
Robert Nishihara
2016-07-08 12:46:47 -07:00
committed by Philipp Moritz
parent 191909dd93
commit 5dd411546d
16 changed files with 81 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/microbenchmarks.py
+2 -3
View File
@@ -1,6 +1,5 @@
import unittest
import ray
import ray.services as services
import time
import os
import numpy as np
@@ -11,7 +10,7 @@ class MicroBenchmarkTest(unittest.TestCase):
def testTiming(self):
worker_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "test_worker.py")
services.start_ray_local(num_workers=3, worker_path=worker_path)
ray.services.start_ray_local(num_workers=3, worker_path=worker_path)
# measure the time required to submit a remote task to the scheduler
elapsed_times = []
@@ -78,7 +77,7 @@ class MicroBenchmarkTest(unittest.TestCase):
print " worst: {}".format(elapsed_times[999])
self.assertTrue(average_elapsed_time < 0.002) # should take 0.00087
services.cleanup()
ray.services.cleanup()
if __name__ == "__main__":
unittest.main()