mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 20:07:41 +08:00
Replace '__main__' with "__main__". (#4055)
This commit is contained in:
committed by
Philipp Moritz
parent
5f71751891
commit
2d07df7f3f
@@ -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()
|
||||
|
||||
@@ -10,7 +10,7 @@ import sys
|
||||
import ray
|
||||
from ray.tune import run_experiments
|
||||
|
||||
if __name__ == '__main__':
|
||||
if __name__ == "__main__":
|
||||
|
||||
ray.init()
|
||||
|
||||
|
||||
@@ -40,5 +40,5 @@ class DQNTest(unittest.TestCase):
|
||||
self.assertNotEqual(r3["evaluation"], r4["evaluation"])
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
if __name__ == "__main__":
|
||||
unittest.main(verbosity=2)
|
||||
|
||||
@@ -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)
|
||||
|
||||
@@ -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)
|
||||
|
||||
@@ -235,5 +235,5 @@ class AsyncSamplesOptimizerTest(unittest.TestCase):
|
||||
raise AssertionError("TIMED OUT", optimizer.stats())
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
if __name__ == "__main__":
|
||||
unittest.main(verbosity=2)
|
||||
|
||||
@@ -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)
|
||||
|
||||
@@ -23,7 +23,7 @@ def easy_objective(config, reporter):
|
||||
time.sleep(0.02)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
if __name__ == "__main__":
|
||||
import argparse
|
||||
|
||||
parser = argparse.ArgumentParser()
|
||||
|
||||
@@ -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()
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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()
|
||||
|
||||
|
||||
@@ -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()
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -23,7 +23,7 @@ def easy_objective(config, reporter):
|
||||
time.sleep(0.02)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
if __name__ == "__main__":
|
||||
import argparse
|
||||
import os
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
@@ -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')
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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')
|
||||
|
||||
@@ -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')
|
||||
|
||||
@@ -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())
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user