diff --git a/python/ray/rllib/optimizers/tests/test_segment_tree.py b/python/ray/rllib/optimizers/tests/test_segment_tree.py index f2a72c648..af53ffa33 100644 --- a/python/ray/rllib/optimizers/tests/test_segment_tree.py +++ b/python/ray/rllib/optimizers/tests/test_segment_tree.py @@ -99,7 +99,7 @@ def test_max_interval_tree(): assert np.isclose(tree.min(3, 4), 3.0) -if __name__ == '__main__': +if __name__ == "__main__": test_tree_set() test_tree_set_overlap() test_prefixsum_idx() diff --git a/python/ray/rllib/test/run_regression_tests.py b/python/ray/rllib/test/run_regression_tests.py index 51be4e4b1..c45cf7b36 100644 --- a/python/ray/rllib/test/run_regression_tests.py +++ b/python/ray/rllib/test/run_regression_tests.py @@ -10,7 +10,7 @@ import sys import ray from ray.tune import run_experiments -if __name__ == '__main__': +if __name__ == "__main__": ray.init() diff --git a/python/ray/rllib/test/test_evaluators.py b/python/ray/rllib/test/test_evaluators.py index ad76c7d4f..6dc532a15 100644 --- a/python/ray/rllib/test/test_evaluators.py +++ b/python/ray/rllib/test/test_evaluators.py @@ -40,5 +40,5 @@ class DQNTest(unittest.TestCase): self.assertNotEqual(r3["evaluation"], r4["evaluation"]) -if __name__ == '__main__': +if __name__ == "__main__": unittest.main(verbosity=2) diff --git a/python/ray/rllib/test/test_external_env.py b/python/ray/rllib/test/test_external_env.py index f7e8308a5..7efdd2744 100644 --- a/python/ray/rllib/test/test_external_env.py +++ b/python/ray/rllib/test/test_external_env.py @@ -195,6 +195,6 @@ class TestExternalEnv(unittest.TestCase): self.assertRaises(ValueError, lambda: ev.sample()) -if __name__ == '__main__': +if __name__ == "__main__": ray.init() unittest.main(verbosity=2) diff --git a/python/ray/rllib/test/test_multi_agent_env.py b/python/ray/rllib/test/test_multi_agent_env.py index ecd957328..f6cf34b29 100644 --- a/python/ray/rllib/test/test_multi_agent_env.py +++ b/python/ray/rllib/test/test_multi_agent_env.py @@ -557,6 +557,6 @@ class TestMultiAgentEnv(unittest.TestCase): raise Exception("failed to improve reward") -if __name__ == '__main__': +if __name__ == "__main__": ray.init() unittest.main(verbosity=2) diff --git a/python/ray/rllib/test/test_optimizers.py b/python/ray/rllib/test/test_optimizers.py index 074e0c081..0bb290018 100644 --- a/python/ray/rllib/test/test_optimizers.py +++ b/python/ray/rllib/test/test_optimizers.py @@ -235,5 +235,5 @@ class AsyncSamplesOptimizerTest(unittest.TestCase): raise AssertionError("TIMED OUT", optimizer.stats()) -if __name__ == '__main__': +if __name__ == "__main__": unittest.main(verbosity=2) diff --git a/python/ray/rllib/test/test_policy_evaluator.py b/python/ray/rllib/test/test_policy_evaluator.py index a71644d26..a810542ad 100644 --- a/python/ray/rllib/test/test_policy_evaluator.py +++ b/python/ray/rllib/test/test_policy_evaluator.py @@ -416,6 +416,6 @@ class TestPolicyEvaluator(unittest.TestCase): return obs_f -if __name__ == '__main__': +if __name__ == "__main__": ray.init(num_cpus=5) unittest.main(verbosity=2) diff --git a/python/ray/tune/examples/bayesopt_example.py b/python/ray/tune/examples/bayesopt_example.py index 06eacd5d0..5f543612f 100644 --- a/python/ray/tune/examples/bayesopt_example.py +++ b/python/ray/tune/examples/bayesopt_example.py @@ -23,7 +23,7 @@ def easy_objective(config, reporter): time.sleep(0.02) -if __name__ == '__main__': +if __name__ == "__main__": import argparse parser = argparse.ArgumentParser() diff --git a/python/ray/tune/examples/genetic_example.py b/python/ray/tune/examples/genetic_example.py index 696a94219..357fcdb61 100644 --- a/python/ray/tune/examples/genetic_example.py +++ b/python/ray/tune/examples/genetic_example.py @@ -27,7 +27,7 @@ def michalewicz_function(config, reporter): reporter(timesteps_total=1, neg_mean_loss=-y) -if __name__ == '__main__': +if __name__ == "__main__": import argparse parser = argparse.ArgumentParser() diff --git a/python/ray/tune/examples/hyperopt_example.py b/python/ray/tune/examples/hyperopt_example.py index 029000e0a..0dc394c2d 100644 --- a/python/ray/tune/examples/hyperopt_example.py +++ b/python/ray/tune/examples/hyperopt_example.py @@ -25,7 +25,7 @@ def easy_objective(config, reporter): time.sleep(0.02) -if __name__ == '__main__': +if __name__ == "__main__": import argparse from hyperopt import hp diff --git a/python/ray/tune/examples/mnist_pytorch.py b/python/ray/tune/examples/mnist_pytorch.py index a1cb983b1..df4072eaf 100644 --- a/python/ray/tune/examples/mnist_pytorch.py +++ b/python/ray/tune/examples/mnist_pytorch.py @@ -150,7 +150,7 @@ def train_mnist(args, config, reporter): test() -if __name__ == '__main__': +if __name__ == "__main__": datasets.MNIST('~/data', train=True, download=True) args = parser.parse_args() diff --git a/python/ray/tune/examples/mnist_pytorch_trainable.py b/python/ray/tune/examples/mnist_pytorch_trainable.py index 52aed6596..d22beddee 100644 --- a/python/ray/tune/examples/mnist_pytorch_trainable.py +++ b/python/ray/tune/examples/mnist_pytorch_trainable.py @@ -165,7 +165,7 @@ class TrainMNIST(Trainable): self.model.load_state_dict(checkpoint_path) -if __name__ == '__main__': +if __name__ == "__main__": datasets.MNIST('~/data', train=True, download=True) args = parser.parse_args() diff --git a/python/ray/tune/examples/nevergrad_example.py b/python/ray/tune/examples/nevergrad_example.py index 0e13e3430..cff8fd795 100644 --- a/python/ray/tune/examples/nevergrad_example.py +++ b/python/ray/tune/examples/nevergrad_example.py @@ -23,7 +23,7 @@ def easy_objective(config, reporter): time.sleep(0.02) -if __name__ == '__main__': +if __name__ == "__main__": import argparse from nevergrad.optimization import optimizerlib diff --git a/python/ray/tune/examples/sigopt_example.py b/python/ray/tune/examples/sigopt_example.py index 8a583e48c..6c18cace6 100644 --- a/python/ray/tune/examples/sigopt_example.py +++ b/python/ray/tune/examples/sigopt_example.py @@ -23,7 +23,7 @@ def easy_objective(config, reporter): time.sleep(0.02) -if __name__ == '__main__': +if __name__ == "__main__": import argparse import os diff --git a/python/ray/tune/examples/skopt_example.py b/python/ray/tune/examples/skopt_example.py index 87cadac6e..71ac60951 100644 --- a/python/ray/tune/examples/skopt_example.py +++ b/python/ray/tune/examples/skopt_example.py @@ -23,7 +23,7 @@ def easy_objective(config, reporter): time.sleep(0.02) -if __name__ == '__main__': +if __name__ == "__main__": import argparse from skopt import Optimizer diff --git a/python/ray/tune/examples/tune_mnist_async_hyperband.py b/python/ray/tune/examples/tune_mnist_async_hyperband.py index 0f8bddc5e..8bc104257 100755 --- a/python/ray/tune/examples/tune_mnist_async_hyperband.py +++ b/python/ray/tune/examples/tune_mnist_async_hyperband.py @@ -212,7 +212,7 @@ def train(config={'activation': 'relu'}, reporter=None): # !!! Example of using the ray.tune Python API !!! -if __name__ == '__main__': +if __name__ == "__main__": parser = argparse.ArgumentParser() parser.add_argument( '--smoke-test', action='store_true', help='Finish quickly for testing') diff --git a/python/ray/tune/examples/tune_mnist_keras.py b/python/ray/tune/examples/tune_mnist_keras.py index 03afe6594..227aee38b 100644 --- a/python/ray/tune/examples/tune_mnist_keras.py +++ b/python/ray/tune/examples/tune_mnist_keras.py @@ -165,7 +165,7 @@ def create_parser(): return parser -if __name__ == '__main__': +if __name__ == "__main__": parser = create_parser() args = parser.parse_args() mnist.load_data() # we do this because it's not threadsafe diff --git a/python/ray/tune/examples/tune_mnist_ray.py b/python/ray/tune/examples/tune_mnist_ray.py index e56ebd10f..a8411ffe7 100755 --- a/python/ray/tune/examples/tune_mnist_ray.py +++ b/python/ray/tune/examples/tune_mnist_ray.py @@ -213,7 +213,7 @@ def train(config={'activation': 'relu'}, reporter=None): # !!! Example of using the ray.tune Python API !!! -if __name__ == '__main__': +if __name__ == "__main__": parser = argparse.ArgumentParser() parser.add_argument( '--smoke-test', action='store_true', help='Finish quickly for testing') diff --git a/python/ray/tune/examples/tune_mnist_ray_hyperband.py b/python/ray/tune/examples/tune_mnist_ray_hyperband.py index 589b53bfe..80a8d97a0 100755 --- a/python/ray/tune/examples/tune_mnist_ray_hyperband.py +++ b/python/ray/tune/examples/tune_mnist_ray_hyperband.py @@ -209,7 +209,7 @@ class TrainMNIST(Trainable): # !!! Example of using the ray.tune Python API !!! -if __name__ == '__main__': +if __name__ == "__main__": parser = argparse.ArgumentParser() parser.add_argument( '--smoke-test', action='store_true', help='Finish quickly for testing') diff --git a/python/ray/tune/schedulers/async_hyperband.py b/python/ray/tune/schedulers/async_hyperband.py index 61e7b8153..156b7d22d 100644 --- a/python/ray/tune/schedulers/async_hyperband.py +++ b/python/ray/tune/schedulers/async_hyperband.py @@ -151,7 +151,7 @@ class _Bracket(): return "Bracket: " + iters -if __name__ == '__main__': +if __name__ == "__main__": sched = AsyncHyperBandScheduler( grace_period=1, max_t=10, reduction_factor=2) print(sched.debug_string()) diff --git a/python/ray/tune/util.py b/python/ray/tune/util.py index fcde28732..ce4047f2e 100644 --- a/python/ray/tune/util.py +++ b/python/ray/tune/util.py @@ -87,7 +87,7 @@ def _from_pinnable(obj): return obj[0] -if __name__ == '__main__': +if __name__ == "__main__": ray.init() X = pin_in_object_store("hello") print(X)