mirror of
https://github.com/wassname/jaxtyping.git
synced 2026-09-09 11:24:55 +08:00
Avoid __builtins__ getting added as a key
This commit is contained in:
@@ -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 "
|
||||
|
||||
Reference in New Issue
Block a user