Set max_total_num_input_blocks to 20, since that is all that's needed.

cubed-dev · Mar 21, 2024 · ae7fe30 · ae7fe30
1 parent a145450
commit ae7fe30
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/benchmarks/test_array.py b/tests/benchmarks/test_array.py
@@ -54,7 +54,7 @@ def test_quadratic_means_xarray(tmp_path, runtime, benchmark_all, optimizer, t_l
     )
 
     if optimizer == "new-optimizer":
-        opt_fn = partial(multiple_inputs_optimize_dag, max_total_num_input_blocks=40)
+        opt_fn = partial(multiple_inputs_optimize_dag, max_total_num_input_blocks=20)
         compute_arrays_in_parallel = True
     else:
         opt_fn = simple_optimize_dag