diff --git a/jaxtyping/import_hook.py b/jaxtyping/import_hook.py index a419fde..31f74e8 100644 --- a/jaxtyping/import_hook.py +++ b/jaxtyping/import_hook.py @@ -104,7 +104,7 @@ class _JaxtypingTransformer(ast.NodeVisitor): 0, ast.Attribute( ast.Name(id="jaxtyping", ctx=ast.Load()), "jaxtyped", ast.Load() - ) + ), ) if self._typechecker is not None: # Place at the end of the decorator list, as decorators