Add message denoting finish of derivative check

chrhansk · Jan 26, 2024 · 64d18a8 · 64d18a8
1 parent bcd69ec
commit 64d18a8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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([])
-Original file line number
+Diff line change
@@ Expand Up @@
                     params,
                 )
+            logger.info("Finished derivative check")
         def print_result(
             self,
             total_time: float,
@@ Expand Down @@