@@ -363,9 +363,9 @@ def test_continue_run_fns(use_emulated_run): # noqa: F811
363
363
364
364
num_steps_orig = len (result_orig .history ["params" ])
365
365
num_steps_full = len (result_full .history ["params" ])
366
- assert (
367
- num_steps_full == num_steps_orig + num_steps_continue
368
- ), "wrong number of elements in the combined run history."
366
+ assert num_steps_full == num_steps_orig + num_steps_continue , (
367
+ "wrong number of elements in the combined run history."
368
+ )
369
369
370
370
371
371
def test_continue_run_from_file (use_emulated_run ): # noqa: F811
@@ -379,17 +379,17 @@ def test_continue_run_from_file(use_emulated_run): # noqa: F811
379
379
)
380
380
num_steps_orig = len (result_orig .history ["params" ])
381
381
num_steps_new = len (result_full .history ["params" ])
382
- assert (
383
- num_steps_new == num_steps_orig + num_steps_continue
384
- ), "wrong number of elements in the combined run history."
382
+ assert num_steps_new == num_steps_orig + num_steps_continue , (
383
+ "wrong number of elements in the combined run history."
384
+ )
385
385
386
386
# test the convenience function to load it from file
387
387
result_full = optimizer .continue_run_from_history (num_steps = 2 , post_process_fn = post_process_fn )
388
388
num_steps_orig = num_steps_new
389
389
num_steps_new = len (result_full .history ["params" ])
390
- assert (
391
- num_steps_new == num_steps_orig + num_steps_continue
392
- ), "wrong number of elements in the combined run history."
390
+ assert num_steps_new == num_steps_orig + num_steps_continue , (
391
+ "wrong number of elements in the combined run history."
392
+ )
393
393
394
394
395
395
def test_result (
@@ -541,8 +541,9 @@ def test_parameter_spec(spec_class, spec_kwargs, expected_shape):
541
541
542
542
543
543
def test_parameter_spec_with_inverse_design (
544
- use_emulated_run , use_emulated_to_sim_data
545
- ): # noqa: F811
544
+ use_emulated_run , # noqa: F811
545
+ use_emulated_to_sim_data ,
546
+ ):
546
547
"""Test InitializationSpec with InverseDesign class."""
547
548
548
549
metric = 2 * ModePower (monitor_name = MNT_NAME2 , f = [FREQ0 ]) ** 2
0 commit comments