From 58340fedfdf586e69d0bc3aa2ff527ac7f2dc8c4 Mon Sep 17 00:00:00 2001 From: Madeline Scyphers Date: Mon, 10 Jul 2023 11:03:32 -0400 Subject: [PATCH] Fix R streamlined test that uses trials key instead of total trials --- tests/integration_tests/test_dunder_main.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/integration_tests/test_dunder_main.py b/tests/integration_tests/test_dunder_main.py index 937e760..a8e625a 100644 --- 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