[tune] Add Unit Test for nested PBT + Jenkins (#7324)

2026-09-12 12:51:15 +08:00 · 2020-02-27 18:17:11 -08:00
parent 8730996682
commit 3fc162f93c
5 changed files with 52 additions and 16 deletions
@@ -48,7 +48,7 @@ if __name__ == "__main__":
    from ray.tune.schedulers import AsyncHyperBandScheduler
    mnist.load_data()  # we do this on the driver because it's not threadsafe

-    ray.init()
+    ray.init(num_cpus=2 if args.smoke_test else None)
    sched = AsyncHyperBandScheduler(
        time_attr="training_iteration",
        metric="mean_accuracy",