mirror of
https://github.com/wassname/jaxtyping.git
synced 2026-09-10 12:14:04 +08:00
Better handling of user-defined typechecker
This commit is contained in:
committed by
Patrick Kidger
parent
9c9635d4f3
commit
513a54b048
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user