chore(deps): bump equinox and undo patch

Author: jeertmans

differt/pyproject.toml

@@ -17,7 +17,7 @@ classifiers = [
 dependencies = [
   "chex>=0.1.86",
   "differt-core==0.0.37",
-  "equinox>=0.11.5",
+  "equinox>=0.12.2",
   "filelock>=3.15.4",
   "jax>=0.4.32",
   "jaxtyping>=0.2.24",

differt/src/differt/conftest.py

@@ -2,7 +2,6 @@
 from typing import Any
 
 import chex
-import equinox as eqx
 import jax
 import jax.numpy as jnp
 import matplotlib.pyplot as plt
@@ -34,19 +33,3 @@ def set_printoptions() -> None:
     # We need to do that because floats seem to vary across OSes:
     # https://github.com/numpy/numpy/issues/21209.
     jax.numpy.set_printoptions(precision=7, suppress=True)
-
-
-class _wrap_method(eqx._module._wrap_method):  # noqa: N801, SLF001
-    def __init__(self, *args: Any, **kwargs: Any) -> None:
-        super().__init__(*args, **kwargs)
-        try:  # noqa: SIM105
-            self.__module__ = self.method.__module__
-        except AttributeError:
-            pass
-        try:  # noqa: SIM105
-            self.__doc__ = self.method.__doc__
-        except AttributeError:
-            pass
-
-
-eqx._module._wrap_method = _wrap_method  # noqa: SLF001