wassname/jaxtyping
1
0
Fork 0
mirror of https://github.com/wassname/jaxtyping.git synced 2026-09-09 11:24:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix abstractmethods being ignored after @jaxtyped

Browse Source
This commit is contained in:
Patrick Kidger
2023-01-22 11:43:36 -08:00
parent f30b7d1546
commit d911ebb99c
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
jaxtyping/decorator.py
+6 -1
View File
@@ -31,7 +31,12 @@ class _Jaxtyped:
self.fn = fn
def __get__(self, instance, owner):
return ft.wraps(self.fn)(_Jaxtyped(self.fn.__get__(instance, owner)))
if instance is None:
# Don't create a new _Jaxtyped object in this case. Otherwise anything
# assigned to methods (e.g. `__isabstractmethod__`) just gets swallowed.
return self
else:
return ft.wraps(self.fn)(_Jaxtyped(self.fn.__get__(instance, owner)))
def __call__(self, *args, **kwargs):
try: