Better handling of user-defined typechecker

This commit is contained in:
Roma Knyaz
2023-10-09 10:21:07 -07:00
committed by Patrick Kidger
parent 9c9635d4f3
commit 513a54b048
5 changed files with 100 additions and 65 deletions
+1 -1
View File
@@ -14,7 +14,7 @@ def ip(session_ip):
session_ip.run_cell(raw_cell="import jaxtyping")
session_ip.run_line_magic(magic_name="load_ext", line="jaxtyping")
session_ip.run_line_magic(
magic_name="jaxtyping.typechecker", line="beartype.beartype"
magic_name="jaxtyping.typechecker", line="typeguard.typechecked"
)
yield session_ip