Make ray test less (not?) stochastic

lanl · Apr 14, 2022 · ef3988b · ef3988b
1 parent ea6dbfb
commit ef3988b
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/scico/test/test_ray_tune.py b/scico/test/test_ray_tune.py
@@ -22,7 +22,10 @@ def eval_params(config, reporter):
 
 tune.ray.tune.register_trainable("eval_func", eval_params)
 
-config = {"x": tune.grid_search(-1, 1), "y": tune.grid_search(-1, 1)}
+config = {
+    "x": tune.grid_search(list(np.linspace(-1, 1, 25))),
+    "y": tune.grid_search(list(np.linspace(-1, 1, 25))),
+}
 resources = {"gpu": 0, "cpu": 1}
 
 
@@ -32,7 +35,7 @@ def test_random():
         "eval_func",
         metric="cost",
         mode="min",
-        num_samples=50,
+        num_samples=1,  # per grid point
         config=config,
         resources_per_trial=resources,
         hyperopt=False,
@@ -50,7 +53,7 @@ def test_hyperopt():
         eval_params,
         metric="cost",
         mode="min",
-        num_samples=50,
+        num_samples=1,  # per grid point
         config=config,
         resources_per_trial=resources,
         hyperopt=True,