`array_typing._check_dataclass_annotations` breaks inside REPLs

[johnnewsom]

if frame.frame.f_globals doesn't have __name__ (e.g. in a REPL), then openpi.shared.array_typing._check_dataclass_annotations raises an error. You can fix this by replacing the raw index (frame.frame.f_globals["__name__"]) with frame.frame.f_globals.get("__name__").

Reproducible by calling Observation.from_dict inside the interpreter

openpi/src/openpi/shared/array_typing.py

Line 34 in 3423421

frame.frame.f_globals["__name__"] in {"jax._src.tree_util", "flax.nnx.transforms.compilation"}