add unroll_nested_dict

daveey · daveey · commit d2a9e8c02b39 · 2025-09-22T23:15:17.000-07:00
diff --git a/packages/pufferlib-core/pufferlib/__init__.py b/packages/pufferlib-core/pufferlib/__init__.py
@@ -13,6 +13,7 @@ def _import_modules():
     current_module = sys.modules[__name__]
     current_module.PufferEnv = pufferlib.PufferEnv
     current_module.set_buffers = pufferlib.set_buffers
+    current_module.unroll_nested_dict = pufferlib.unroll_nested_dict
 
     from . import emulation
     from . import vector
