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

Fixes #188.

Browse Source
This commit is contained in:
Patrick Kidger
2024-03-07 18:28:08 +01:00
parent f708d1742b
commit 5f509adefb
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
jaxtyping/_decorator.py
+5 -1
View File
@@ -189,7 +189,11 @@ def jaxtyped(fn=_sentinel, *, typechecker=_sentinel):
"""
global _tb_flag
if _tb_flag and importlib.util.find_spec("jax._src.traceback_util") is not None:
if (
_tb_flag
and importlib.util.find_spec("jax") is not None
and importlib.util.find_spec("jax._src.traceback_util") is not None
):
import jax._src.traceback_util as traceback_util
traceback_util.register_exclusion(__file__)
pyproject.toml
+1 -1
View File
@@ -1,6 +1,6 @@
[project]
name = "jaxtyping"
version = "0.2.27"
version = "0.2.28"
description = "Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees."
readme = "README.md"
requires-python ="~=3.9"