From 64d18a80f3c70ca5a79e6336858fae1a6dd1b072 Mon Sep 17 00:00:00 2001 From: Christoph Hansknecht Date: Fri, 26 Jan 2024 20:15:29 +0100 Subject: [PATCH] Add message denoting finish of derivative check --- pygradflow/solver.py | 2 ++ tests/pygradflow/rosenbrock.py | 4 +--- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/pygradflow/solver.py b/pygradflow/solver.py index 58bfffc..ed277ec 100644 --- a/pygradflow/solver.py +++ b/pygradflow/solver.py @@ -254,6 +254,8 @@ def _deriv_check(self, x: np.ndarray, y: np.ndarray) -> None: params, ) + logger.info("Finished derivative check") + def print_result( self, total_time: float, diff --git a/tests/pygradflow/rosenbrock.py b/tests/pygradflow/rosenbrock.py index e658946..31d085e 100644 --- a/tests/pygradflow/rosenbrock.py +++ b/tests/pygradflow/rosenbrock.py @@ -23,9 +23,7 @@ def obj_grad(self, v): [x, y] = v a = self.a b = self.b - return np.array( - [4 * (x**2 - y) * b * x - 2 * a + 2 * x, -2 * (x**2 - y) * b] - ) + return np.array([4 * (x**2 - y) * b * x - 2 * a + 2 * x, -2 * (x**2 - y) * b]) def cons(self, v): return np.array([])