diff --git a/equinox/_module.py b/equinox/_module.py index 7b0b5cf1..f1b15d0b 100644 --- a/equinox/_module.py +++ b/equinox/_module.py @@ -101,7 +101,9 @@ def __init__(self, method): def __get__(self, instance, owner): if instance is None: return self.method - return jtu.Partial(self.method, instance) + _method = ft.wraps(self.method)(jtu.Partial(self.method, instance)) + delattr(_method, "__wrapped__") + return _method def _not_magic(k: str) -> bool: