Delete dead code

Author: pnkraemer

probdiffeq/impl/_normal.py

@@ -60,7 +60,7 @@ def from_tcoeffs(self, tcoeffs: list, damp: float = 0.0):
         powers = 1 / np.arange(1, len(tcoeffs) + 1)
         c_sqrtm0_corrected = linalg.diagonal_matrix(damp**powers)
 
-        leaves, structure = tree_util.tree_flatten(tcoeffs)
+        leaves, _ = tree_util.tree_flatten(tcoeffs)
         m0_corrected = np.stack(leaves)
         return Normal(m0_corrected, c_sqrtm0_corrected)
 
@@ -82,7 +82,7 @@ def from_tcoeffs(self, tcoeffs: list, damp: float = 0.0):
         cholesky = linalg.diagonal_matrix(damp**powers)
         cholesky = np.ones((*self.ode_shape, 1, 1)) * cholesky[None, ...]
 
-        leaves, structure = tree_util.tree_flatten(tcoeffs)
+        leaves, _ = tree_util.tree_flatten(tcoeffs)
         mean = np.stack(leaves).T
         return Normal(mean, cholesky)
 

probdiffeq/ivpsolvers.py

@@ -441,24 +441,22 @@ class _Correction:
     linearize: Callable
     vector_field: Callable
 
-    can_handle_higher_order: bool
-
     def init(self, x, /):
         """Initialise the state from the solution."""
         y = self.ssm.prototypes.observed()
         return x, y
 
     def correct(self, rv, /, t):
         """Perform the correction step."""
-        f_wrapped = self._parametrize_vector_field(t=t)
+        f_wrapped = functools.partial(self.vector_field, t=t)
         cond = self.linearize(f_wrapped, rv)
         observed, reverted = self.ssm.conditional.revert(rv, cond)
         corrected = reverted.noise
         return corrected, observed
 
     def estimate_error(self, rv, /, t):
         """Estimate the error."""
-        f_wrapped = self._parametrize_vector_field(t=t)
+        f_wrapped = functools.partial(self.vector_field, t=t)
         cond = self.linearize(f_wrapped, rv)
         observed = self.ssm.conditional.marginalise(rv, cond)
 
@@ -469,9 +467,6 @@ def estimate_error(self, rv, /, t):
         error_estimate = output_scale * error_estimate_unscaled
         return error_estimate, observed
 
-    def _parametrize_vector_field(self, *, t):
-        return functools.partial(self.vector_field, t=t)
-
 
 def correction_ts0(vector_field, *, ssm, ode_order=1, damp: float = 0.0) -> _Correction:
     """Zeroth-order Taylor linearisation."""
@@ -482,7 +477,6 @@ def correction_ts0(vector_field, *, ssm, ode_order=1, damp: float = 0.0) -> _Cor
         ode_order=ode_order,
         ssm=ssm,
         linearize=linearize,
-        can_handle_higher_order=True,
     )
 
 
@@ -495,7 +489,6 @@ def correction_ts1(vector_field, *, ssm, ode_order=1, damp: float = 0.0) -> _Cor
         ode_order=ode_order,
         ssm=ssm,
         linearize=linearize,
-        can_handle_higher_order=True,
     )
 
 
@@ -510,7 +503,6 @@ def correction_slr0(
         ode_order=1,
         linearize=linearize,
         name="SLR0",
-        can_handle_higher_order=False,  # TODO: implement this
     )
 
 
@@ -525,7 +517,6 @@ def correction_slr1(
         ode_order=1,
         linearize=linearize,
         name="SLR1",
-        can_handle_higher_order=False,  # TODO: implement this
     )