pyglove 0.5.0.dev202507240811__py3-none-any.whl → 0.5.0.dev202507260809__py3-none-any.whl

pyglove/core/utils/json_conversion.py

@@ -23,7 +23,7 @@ import marshal
 import pickle
 import types
 import typing
-from typing import Any, Callable, ContextManager, Dict, Iterable, Iterator, List, Optional, Set, Tuple, Type, TypeVar, Union
+from typing import Any, Callable, ContextManager, Dict, Iterable, Iterator, List, Optional, Sequence, Set, Tuple, Type, TypeVar, Union
 
 # Nestable[T] is a (maybe) nested structure of T, which could be T, a Dict
 # a List or a Tuple of Nestable[T]. We use a Union to fool PyType checker to
@@ -83,9 +83,16 @@ class _TypeRegistry:
           f'{self._type_to_cls_map[type_name].__name__}.')
     self._type_to_cls_map[type_name] = cls
 
-  def add_module_alias(self, module: str, alias: str) -> None:
+  def add_module_alias(
+      self,
+      module: str,
+      alias: Union[str, Sequence[str]]
+  ) -> None:
     """Maps a module name to another name. Usually due to rename."""
-    self._prefix_mapping[alias] = module
+    if isinstance(alias, str):
+      alias = [alias]
+    for name in alias:
+      self._prefix_mapping[name] = module
 
   def is_registered(self, type_name: str) -> bool:
     """Returns whether a type name is registered."""
@@ -245,9 +252,13 @@ class JSONConvertible(metaclass=abc.ABCMeta):
     cls._TYPE_REGISTRY.register(type_name, subclass, override_existing)
 
   @classmethod
-  def add_module_alias(cls, source_name: str, target_name: str) -> None:
+  def add_module_alias(
+      cls,
+      module: str,
+      alias: Union[str, Sequence[str]]
+  ) -> None:
     """Adds a module alias so previous serialized objects could be loaded."""
-    cls._TYPE_REGISTRY.add_module_alias(source_name, target_name)
+    cls._TYPE_REGISTRY.add_module_alias(module, alias)
 
   @classmethod
   def is_registered(cls, type_name: str) -> bool:

{pyglove-0.5.0.dev202507240811.dist-info → pyglove-0.5.0.dev202507260809.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyglove
-Version: 0.5.0.dev202507240811
+Version: 0.5.0.dev202507260809
 Summary: PyGlove: A library for manipulating Python objects.
 Home-page: https://github.com/google/pyglove
 Author: PyGlove Authors

{pyglove-0.5.0.dev202507240811.dist-info → pyglove-0.5.0.dev202507260809.dist-info}/RECORD

@@ -147,7 +147,7 @@ pyglove/core/utils/formatting.py,sha256=Wn4d933LQLhuMIfjdRJgpxOThCxBxQrkRBa6Z1-h
 pyglove/core/utils/formatting_test.py,sha256=hhg-nL6DyE5A2QA92ALHK5QtfAYKfPpTbBARF-IT1j0,14241
 pyglove/core/utils/hierarchical.py,sha256=jwB-0FhqOspAymAkvJphRhPTQEsoShmKupCZpU3Vip4,19690
 pyglove/core/utils/hierarchical_test.py,sha256=f382DMJPa_bavJGGQDjuw-hWcafUg5bkQCPX-nbzeiI,21077
-pyglove/core/utils/json_conversion.py,sha256=fJ3zVvXJaoBf2OgUMYuYiSbRWuFg2PY82Vqzf9YohUc,26632
+pyglove/core/utils/json_conversion.py,sha256=26Wt7VDU4df6fT_Joly-UzU87abkAQNL8gYDIh36fQ8,26785
 pyglove/core/utils/json_conversion_test.py,sha256=zA_cy7ixVL3sTf6i9BCXMlSH56Aa3JnjHnjyqYJ_9XU,11845
 pyglove/core/utils/missing.py,sha256=9gslt1lXd1qSEIuAFxUWu30oD-YdYcnm13eau1S9uqY,1445
 pyglove/core/utils/missing_test.py,sha256=D6-FuVEwCyJemUiPLcwLmwyptqI5Bx0Pfipc2juhKSE,1335
@@ -216,8 +216,8 @@ pyglove/ext/scalars/randoms.py,sha256=LkMIIx7lOq_lvJvVS3BrgWGuWl7Pi91-lA-O8x_gZs
 pyglove/ext/scalars/randoms_test.py,sha256=nEhiqarg8l_5EOucp59CYrpO2uKxS1pe0hmBdZUzRNM,2000
 pyglove/ext/scalars/step_wise.py,sha256=IDw3tuTpv0KVh7AN44W43zqm1-E0HWPUlytWOQC9w3Y,3789
 pyglove/ext/scalars/step_wise_test.py,sha256=TL1vJ19xVx2t5HKuyIzGoogF7N3Rm8YhLE6JF7i0iy8,2540
-pyglove-0.5.0.dev202507240811.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-pyglove-0.5.0.dev202507240811.dist-info/METADATA,sha256=_GS4joAJ4qJ_2OFrIQ4gC--ML18KWBGqi_2in_-nZzs,7089
-pyglove-0.5.0.dev202507240811.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pyglove-0.5.0.dev202507240811.dist-info/top_level.txt,sha256=wITzJSKcj8GZUkbq-MvUQnFadkiuAv_qv5qQMw0fIow,8
-pyglove-0.5.0.dev202507240811.dist-info/RECORD,,
+pyglove-0.5.0.dev202507260809.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+pyglove-0.5.0.dev202507260809.dist-info/METADATA,sha256=q-Ed_FCh7XUuuDaDhO0V4c9q-gtjvb9uuctIv2O1C40,7089
+pyglove-0.5.0.dev202507260809.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pyglove-0.5.0.dev202507260809.dist-info/top_level.txt,sha256=wITzJSKcj8GZUkbq-MvUQnFadkiuAv_qv5qQMw0fIow,8
+pyglove-0.5.0.dev202507260809.dist-info/RECORD,,