diff --git a/jaxtyping/array_types.py b/jaxtyping/array_types.py index 45cc969..489efbe 100644 --- a/jaxtyping/array_types.py +++ b/jaxtyping/array_types.py @@ -123,7 +123,8 @@ def _check_dims( return False elif type(cls_dim) is _SymbolicDim: try: - eval_size = eval(cls_dim.expr, single_memo) + # Make a copy to avoid `__builtins__` getting added as a key. + eval_size = eval(cls_dim.expr, single_memo.copy()) except NameError as e: raise NameError( f"Cannot process symbolic dimension '{cls_dim.expr}' as some "