Fix R streamlined test that uses trials key instead of total trials

madeline-scyphers · Jul 10, 2023 · 58340fe · 58340fe · github-actions · Jul 10, 2023
1 parent 942aaa5
commit 58340fe
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/integration_tests/test_dunder_main.py b/tests/integration_tests/test_dunder_main.py
@@ -47,7 +47,9 @@ def test_calling_command_line_r_test_scripts(r_scripts_run):
     scheduler = r_scripts_run
     wrapper = scheduler.experiment.runner.wrapper
     config = wrapper.config
-    total_trials = config["optimization_options"]["scheduler"]["total_trials"]
+    total_trials = config["optimization_options"].get("scheduler", {}).get("total_trials") or config[
+        "optimization_options"
+    ].get("trials")
     assert len(scheduler.experiment.trials) == total_trials