From 43c3cf8680fce4cf33c7b19090f843c496a2e660 Mon Sep 17 00:00:00 2001 From: John Gerrard Holland Date: Fri, 7 Jul 2023 17:20:16 -0400 Subject: [PATCH] test: update testcases to use num_sample rather than n --- src/autora/workflow/__init__.py | 2 +- tests/test_controller_plots.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/autora/workflow/__init__.py b/src/autora/workflow/__init__.py index 53957f63..9444603b 100644 --- a/src/autora/workflow/__init__.py +++ b/src/autora/workflow/__init__.py @@ -464,7 +464,7 @@ Wen can run the seed pipeline with no data: >>> experimentalist_which_needs_no_data = make_pipeline([ ... np.linspace(*variables_2.independent_variables[0].value_range, 1_000), - ... partial(random_sampler, n=10)] + ... partial(random_sampler, num_samples=10)] ... ) >>> np.array(experimentalist_which_needs_no_data()) array([ 6.71671672, -0.73073073, -5.05505506, 6.13613614, 0.03003003, diff --git a/tests/test_controller_plots.py b/tests/test_controller_plots.py index 8d7b90df..2a5da2ea 100644 --- a/tests/test_controller_plots.py +++ b/tests/test_controller_plots.py @@ -52,7 +52,7 @@ def state_lr(ground_truth_1x): ], params={ "pool": {"ivs": study_variables.independent_variables}, - "sampler": {"n": 5}, + "sampler": {"num_samples": 5}, }, ) @@ -117,7 +117,7 @@ def cycle_multi_lr(ground_truth_2x): ], params={ "pool": {"ivs": study_variables.independent_variables}, - "sampler": {"n": 10}, + "sampler": {"num_samples": 10}, }, )