Merge pull request #33 from AutoResearch/fix-broken-tests-random_sample

test: fix broken tests (random_sample uses num_samples now)
AutoResearch · Jul 7, 2023 · 7622917 · 7622917
2 parents df4742a + 43c3cf8
commit 7622917
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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},
         },
     )