diff --git a/pyciemss/ouu/ouu.py b/pyciemss/ouu/ouu.py index 962511fd..2390b67f 100644 --- a/pyciemss/ouu/ouu.py +++ b/pyciemss/ouu/ouu.py @@ -185,7 +185,9 @@ def __init__( maxfeval: int = 100, maxiter: int = 100, u_bounds: np.ndarray = np.atleast_2d([[0], [1]]), - progress_hook: Callable[[np.array, float, bool], bool] = lambda x, f, accept: False, + progress_hook: Callable[ + [np.array, float, bool], bool + ] = lambda x, f, accept: False, ): self.x0 = np.squeeze(np.array([x0])) self.objfun = objfun diff --git a/tests/test_interfaces.py b/tests/test_interfaces.py index d65e53d5..3db6e496 100644 --- a/tests/test_interfaces.py +++ b/tests/test_interfaces.py @@ -634,9 +634,7 @@ def __call__(self, x, f, accept): intervened_result_subset, start_time, end_time, logging_step_size, num_samples ) - assert len(progress_hook.coordinates) == ( - optimize_kwargs["maxiter"] + 1 - ) + assert len(progress_hook.coordinates) == (optimize_kwargs["maxiter"] + 1) @pytest.mark.parametrize("model_fixture", MODELS)