Skip to content

Commit 9b788ca

Browse files
committed
fix test
1 parent dec3c47 commit 9b788ca

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

neps_examples/convenience/async_evaluation/submit.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,5 @@ def evaluate_pipeline_via_slurm(pipeline_id, pipeline_directory, previous_pipeli
4343
evaluate_pipeline=evaluate_pipeline_via_slurm,
4444
pipeline_space=pipeline_space,
4545
root_directory="results",
46-
max_evaluations_total=2,
47-
post_run_summary=True,
46+
max_evaluations_per_run=2,
4847
)

tests/test_runtime/test_save_evaluation_results.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -38,15 +38,16 @@ def test_async_happy_path_changes_state(neps_state: NePSState) -> None:
3838
default_report_values=DefaultReportValues(
3939
cost_if_not_provided=1.2,
4040
),
41-
max_evaluations_total=None,
42-
include_in_progress_evaluations_towards_maximum=False,
43-
max_cost_total=None,
41+
cost_to_spend=None,
4442
max_evaluations_for_worker=2,
4543
max_evaluation_time_total_seconds=None,
4644
max_wallclock_time_for_worker_seconds=None,
4745
max_evaluation_time_for_worker_seconds=None,
4846
max_cost_for_worker=None,
4947
batch_size=None,
48+
fidelities_to_spend=None,
49+
evaluations_to_spend=1,
50+
include_in_progress_evaluations_towards_maximum=False,
5051
)
5152

5253
callback_holder: list[callable] = []

0 commit comments

Comments
 (0)