diff --git a/examples/hyperopt/worker.py b/examples/hyperopt/worker.py index 357d5345e..8df63e252 100644 --- a/examples/hyperopt/worker.py +++ b/examples/hyperopt/worker.py @@ -9,7 +9,7 @@ parser.add_argument("--scheduler-address", default="127.0.0.1:10001", type=str, parser.add_argument("--objstore-address", default="127.0.0.1:20001", type=str, help="the objstore's address") parser.add_argument("--worker-address", default="127.0.0.1:40001", type=str, help="the worker's address") -if __name__ == '__main__': +if __name__ == "__main__": args = parser.parse_args() ray.connect(args.scheduler_address, args.objstore_address, args.worker_address) ray.register_module(functions) diff --git a/examples/rl_pong/worker.py b/examples/rl_pong/worker.py index 9c3b26510..196dbff87 100644 --- a/examples/rl_pong/worker.py +++ b/examples/rl_pong/worker.py @@ -10,7 +10,7 @@ parser.add_argument("--scheduler-address", default="127.0.0.1:10001", type=str, parser.add_argument("--objstore-address", default="127.0.0.1:20001", type=str, help="the objstore's address") parser.add_argument("--worker-address", default="127.0.0.1:40001", type=str, help="the worker's address") -if __name__ == '__main__': +if __name__ == "__main__": args = parser.parse_args() ray.connect(args.scheduler_address, args.objstore_address, args.worker_address) ray.register_module(functions) diff --git a/scripts/shell.py b/scripts/shell.py index c867efb1a..9f708d4d2 100644 --- a/scripts/shell.py +++ b/scripts/shell.py @@ -13,7 +13,7 @@ parser.add_argument("--scheduler-address", default="127.0.0.1:10001", type=str, parser.add_argument("--objstore-address", default="127.0.0.1:20001", type=str, help="the objstore's address") parser.add_argument("--worker-address", default="127.0.0.1:30001", type=str, help="the worker's address") -if __name__ == '__main__': +if __name__ == "__main__": args = parser.parse_args() worker.connect(args.scheduler_address, args.objstore_address, args.worker_address) diff --git a/test/arrays_test.py b/test/arrays_test.py index 12b1cf537..a1e39732d 100644 --- a/test/arrays_test.py +++ b/test/arrays_test.py @@ -208,5 +208,5 @@ class ArraysDistTest(unittest.TestCase): services.cleanup() -if __name__ == '__main__': +if __name__ == "__main__": unittest.main() diff --git a/test/microbenchmarks.py b/test/microbenchmarks.py index 22ed29035..c41f40591 100644 --- a/test/microbenchmarks.py +++ b/test/microbenchmarks.py @@ -80,5 +80,5 @@ class MicroBenchmarkTest(unittest.TestCase): services.cleanup() -if __name__ == '__main__': +if __name__ == "__main__": unittest.main() diff --git a/test/runtest.py b/test/runtest.py index da176e70d..99692b4ab 100644 --- a/test/runtest.py +++ b/test/runtest.py @@ -65,14 +65,14 @@ class SerializationTest(unittest.TestCase): b = serialization.deserialize(w.handle, res) self.assertTrue((a == b).all()) - self.numpyTypeTest(w, 'int8') - self.numpyTypeTest(w, 'uint8') - self.numpyTypeTest(w, 'int16') - self.numpyTypeTest(w, 'uint16') - self.numpyTypeTest(w, 'int32') - self.numpyTypeTest(w, 'uint32') - self.numpyTypeTest(w, 'float32') - self.numpyTypeTest(w, 'float64') + self.numpyTypeTest(w, "int8") + self.numpyTypeTest(w, "uint8") + self.numpyTypeTest(w, "int16") + self.numpyTypeTest(w, "uint16") + self.numpyTypeTest(w, "int32") + self.numpyTypeTest(w, "uint32") + self.numpyTypeTest(w, "float32") + self.numpyTypeTest(w, "float64") ref0 = ray.put(0, w) ref1 = ray.put(0, w) @@ -86,7 +86,7 @@ class SerializationTest(unittest.TestCase): self.roundTripTest(w, ref0) self.roundTripTest(w, [ref0, ref1, ref2, ref3]) - self.roundTripTest(w, {'0': ref0, '1': ref1, '2': ref2, '3': ref3}) + self.roundTripTest(w, {"0": ref0, "1": ref1, "2": ref2, "3": ref3}) self.roundTripTest(w, (ref0, 1)) services.cleanup() @@ -247,14 +247,14 @@ class TaskStatusTest(unittest.TestCase): test_functions.throw_exception_fct() time.sleep(1) result = ray.task_info() - self.assertTrue(len(result['failed_tasks']) == 2) + self.assertTrue(len(result["failed_tasks"]) == 2) task_ids = set() - for task in result['failed_tasks']: - self.assertTrue(task.has_key('worker_address')) - self.assertTrue(task.has_key('operationid')) - self.assertEqual(task.get('error_message'), "Test function intentionally failed.") - self.assertTrue(task['operationid'] not in task_ids) - task_ids.add(task['operationid']) + for task in result["failed_tasks"]: + self.assertTrue(task.has_key("worker_address")) + self.assertTrue(task.has_key("operationid")) + self.assertEqual(task.get("error_message"), "Test function intentionally failed.") + self.assertTrue(task["operationid"] not in task_ids) + task_ids.add(task["operationid"]) def check_get_deallocated(data): x = ray.put(data) @@ -360,5 +360,5 @@ class ReferenceCountingTest(unittest.TestCase): services.cleanup() -if __name__ == '__main__': +if __name__ == "__main__": unittest.main() diff --git a/test/test_worker.py b/test/test_worker.py index 2b02803d0..0310ae76e 100644 --- a/test/test_worker.py +++ b/test/test_worker.py @@ -16,7 +16,7 @@ parser.add_argument("--scheduler-address", default="127.0.0.1:10001", type=str, parser.add_argument("--objstore-address", default="127.0.0.1:20001", type=str, help="the objstore's address") parser.add_argument("--worker-address", default="127.0.0.1:40001", type=str, help="the worker's address") -if __name__ == '__main__': +if __name__ == "__main__": args = parser.parse_args() worker.connect(args.scheduler_address, args.objstore_address, args.worker_address)