pyglove 0.4.5.dev202504210810__py3-none-any.whl → 0.4.5.dev202504230810__py3-none-any.whl

pyglove/core/coding/errors.py

@@ -28,7 +28,7 @@ class CodeError(RuntimeError):
   def __init__(
       self,
       code: str,
-      cause: Exception,
+      cause: BaseException,
       ):
     self.code = code
     self.cause = cause
@@ -93,7 +93,7 @@ class CodeError(RuntimeError):
 class SerializationError(RuntimeError):
   """Object serialization error."""
 
-  def __init__(self, message: Optional[str], cause: Exception):
+  def __init__(self, message: Optional[str], cause: BaseException):
     self.message = message
     self.cause = cause
 

pyglove/core/coding/execution.py

@@ -131,7 +131,7 @@ def evaluate(
         result = eval(  # pylint: disable=eval-used
             compile(last_expr, '', mode='eval'), global_vars
         )
-      except Exception as e:
+      except BaseException as e:
         raise errors.CodeError(code, e) from e
 
       for result_var in result_vars:
@@ -139,7 +139,7 @@ def evaluate(
     else:
       try:
         exec(compile(code_block, '', mode='exec'), global_vars)  # pylint: disable=exec-used
-      except Exception as e:
+      except BaseException as e:
         raise errors.CodeError(code, e) from e
       global_vars[RESULT_KEY] = list(global_vars.values())[-1]
 
@@ -188,7 +188,7 @@ def sandbox_call(
       r = func(*args, **kwargs)
       try:
         return pickle.dumps(r)
-      except Exception as e:
+      except BaseException as e:
         raise errors.SerializationError(
             f'Cannot serialize sandbox result: {r}', e
         ) from e

pyglove/core/coding/execution_test.py

@@ -217,6 +217,10 @@ class EvaluateTest(unittest.TestCase):
       execution.evaluate(
           'raise ValueError()', permission=permissions.CodePermission.ALL
       )
+    with self.assertRaisesRegex(errors.CodeError, 'SystemExit'):
+      execution.evaluate(
+          'raise SystemExit()', permission=permissions.CodePermission.ALL
+      )
 
 
 @dataclasses.dataclass

pyglove/core/utils/formatting.py

@@ -161,6 +161,7 @@ def kvlist_str(
     label: Optional[str] = None,
     bracket_type: BracketType = BracketType.ROUND,
     custom_format: Optional[CustomFormatFn] = None,
+    memo: Optional[Set[int]] = None,
     **kwargs,
 ) -> str:
   """Formats a list key/value pairs into a comma delimited string.
@@ -178,6 +179,8 @@ def kvlist_str(
     custom_format: An optional custom format function, which will be applied to
       each value (and child values) in kvlist. If the function returns None, it
       will fall back to the default `pg.format`.
+    memo: A set of object ids that have been formatted. Used to avoid
+      infinite recursion in the formatting process.
     **kwargs: Keyword arguments that will be passed through unto child
       ``Formattable`` objects.
   Returns:
@@ -208,6 +211,7 @@ def kvlist_str(
           verbose=verbose,
           root_indent=child_indent,
           custom_format=custom_format,
+          memo=memo,
           **kwargs
       )
       if not compact:
@@ -271,6 +275,7 @@ def format(   # pylint: disable=redefined-builtin
     max_bytes_len: Optional[int] = None,
     *,
     custom_format: Optional[CustomFormatFn] = None,
+    memo: Optional[Set[int]] = None,
     **kwargs,
 ) -> str:
   """Formats a (maybe) hierarchical value with flags.
@@ -297,6 +302,8 @@ def format(   # pylint: disable=redefined-builtin
     custom_format: An optional custom format function, which will be applied to
       each value (and child values) in kvlist. If the function returns None, it
       will fall back to the default `pg.format`.
+    memo: A set of object ids that have been formatted. Used to avoid
+      infinite recursion in the formatting process.
     **kwargs: Keyword arguments that will be passed through unto child
       ``Formattable`` objects.
 
@@ -309,6 +316,14 @@ def format(   # pylint: disable=redefined-builtin
     if result is not None:
       return maybe_markdown_quote(result, markdown)
 
+  if memo is None:
+    memo = set()
+
+  id_ = id(value)
+  if id_ in memo:
+    return f'<recursive {value.__class__.__name__} at 0x{id_:x}>'  # pylint: disable=bad-whitespace
+  memo.add(id_)
+
   exclude_keys = exclude_keys or set()
 
   def _should_include_key(key: str) -> bool:
@@ -327,6 +342,7 @@ def format(   # pylint: disable=redefined-builtin
         max_str_len=max_str_len,
         max_bytes_len=max_bytes_len,
         custom_format=custom_format,
+        memo=memo,
         **kwargs
     )
 
@@ -345,6 +361,7 @@ def format(   # pylint: disable=redefined-builtin
           max_str_len=max_str_len,
           max_bytes_len=max_bytes_len,
           custom_format=custom_format,
+          memo=memo,
           **kwargs
       )
     elif isinstance(value, (list, tuple)):
@@ -393,6 +410,7 @@ def format(   # pylint: disable=redefined-builtin
              if compact else str_ext(value, custom_format, root_indent)]
         if strip_object_id and 'object at 0x' in s[-1]:
           s = [f'{value.__class__.__name__}(...)']
+  memo.remove(id_)
   return maybe_markdown_quote(''.join(s), markdown)
 
 
@@ -454,10 +472,10 @@ def camel_to_snake(text: str, separator: str = '_') -> str:
   return (separator.join(c for c in chunks if c)).lower()
 
 
-def printv(v: Any, **kwargs):
+def printv(*args, **kwargs):
   """Prints formatted value."""
   fs = kwargs.pop('file', sys.stdout)
-  print(format(v, **kwargs), file=fs)
+  print(*[format(v, **kwargs) for v in args], file=fs)
 
 
 def _indent(text: str, indent: int) -> str:

pyglove/core/utils/formatting_test.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import inspect
+import io
 import unittest
 
 from pyglove.core.utils import formatting
@@ -408,6 +409,26 @@ class FormatTest(unittest.TestCase):
         '{\'x\': <a/>}'
     )
 
+  def test_recursion(self):
+    # Recursive dict.
+    x = dict(x=1)
+    x['y'] = x
+    self.assertEqual(
+        formatting.format(x, compact=True),
+        f'{{\'x\': 1, \'y\': <recursive dict at 0x{id(x):x}>}}'
+    )
+    self.assertEqual(
+        formatting.format(x, compact=False),
+        f'{{\n  \'x\': 1,\n  \'y\': <recursive dict at 0x{id(x):x}>\n}}'
+    )
+    # Non-recursive dict.
+    y = dict(x=1)
+    a = dict(x=y, y=y)
+    self.assertEqual(
+        formatting.format(a, compact=True),
+        '{\'x\': {\'x\': 1}, \'y\': {\'x\': 1}}'
+    )
+
   def test_markdown(self):
 
     class A(formatting.Formattable):
@@ -448,6 +469,11 @@ class FormatTest(unittest.TestCase):
         formatting.format(b'bar', max_bytes_len=2), 'b\'ba...\''
     )
 
+  def test_printv(self):
+    with io.StringIO() as f:
+      formatting.printv(1, 2, 3, file=f)
+      self.assertEqual(f.getvalue(), '1 2 3\n')
+
 
 if __name__ == '__main__':
   unittest.main()

{pyglove-0.4.5.dev202504210810.dist-info → pyglove-0.4.5.dev202504230810.dist-info}/METADATA

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

{pyglove-0.4.5.dev202504210810.dist-info → pyglove-0.4.5.dev202504230810.dist-info}/RECORD

@@ -3,10 +3,10 @@ pyglove/core/__init__.py,sha256=BF-8v1D5lTBAo4vlpDt0qyOG9NQ71oh_UfthBVrPaN4,9614
 pyglove/core/logging.py,sha256=Sr5nLyUQmc4CAEYAq8qbP3ghpjmpz2sOuhq2A0tgQ6I,2456
 pyglove/core/logging_test.py,sha256=3z_c6wnxbqDbwUmSOAZzeDPXvzoweYL5QHUQVMJ5Xgo,1917
 pyglove/core/coding/__init__.py,sha256=tuHIg19ZchtkOQbdFVTVLkUpBa5f1eo66XtnKw3lcIU,1645
-pyglove/core/coding/errors.py,sha256=vstDb52lk5uJef2k_0_1dqvGv8eDYNq37IrqjGJ-cNc,3302
+pyglove/core/coding/errors.py,sha256=aP3Y4amBzOKdlb5JnESJ3kdoijQXbiBiPDMeA88LNrk,3310
 pyglove/core/coding/errors_test.py,sha256=fwOR8vLiRvLadubsccyE19hLHj-kThlCQt88qmUYk9M,2311
-pyglove/core/coding/execution.py,sha256=Fo5PS8FrKNxAQmzmyZvv2E7MabgcGsnGUw9YIhFEBgg,10931
-pyglove/core/coding/execution_test.py,sha256=ghlwjNgL14-4_ZRDA8SmwQJzjOa2zhb4y6N6gV1DWso,8938
+pyglove/core/coding/execution.py,sha256=KGKl4ovtAPYJI-eQTLBTMwqE4FXSLbTDCj5NI4kVmy4,10943
+pyglove/core/coding/execution_test.py,sha256=7V6hwShKiqpLR8jZl1tFvV3qV8QW5ZcEtBZ1rSOFPrc,9111
 pyglove/core/coding/function_generation.py,sha256=2fCqcdcizVYtGYE6QGpw5m1kuH9Fp0OF4BjyJ4en6tw,1636
 pyglove/core/coding/function_generation_test.py,sha256=kbSwmZF8Vog0R0OTSpuzPblEbMLoRJ1TigeIrwDhHS8,2161
 pyglove/core/coding/parsing.py,sha256=JXTdzFS9iB5-JVi9_-rDFORj_EZEtr7In8HBGHp09no,4084
@@ -139,8 +139,8 @@ pyglove/core/utils/docstr_utils.py,sha256=5BY40kXozPKVGOB0eN8jy1P5_GHIzqFJ9FXAu_
 pyglove/core/utils/docstr_utils_test.py,sha256=i33VT6zHXEuIIJ4PPg1bVSfaYSnUsC8yST_NfpT-Uds,4228
 pyglove/core/utils/error_utils.py,sha256=ACIqtq4hsWrV00Gxd1OyeHCI-obGBR6J3H9VI1Y9clM,5780
 pyglove/core/utils/error_utils_test.py,sha256=zwTzmyJupIW8GeF3Z-gnpVNdhdTYwzmO0ClODqbR4bU,4164
-pyglove/core/utils/formatting.py,sha256=qa5XEGRwKEEqEKtyEYZNdsKBxIQFGBTofFW29aQT8hU,14987
-pyglove/core/utils/formatting_test.py,sha256=OQboyCwhgvhw4EcCj1q-O53XG404c0XMK51rCl2_q2M,13533
+pyglove/core/utils/formatting.py,sha256=Wn4d933LQLhuMIfjdRJgpxOThCxBxQrkRBa6Z1-hL_I,15591
+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=I0mWn87aAEdaAok9nDvT0ZrmplU40eNmEDUAaNIzZXk,26590
@@ -212,8 +212,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.4.5.dev202504210810.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-pyglove-0.4.5.dev202504210810.dist-info/METADATA,sha256=u4BueJVOwhXWVMn51RnQHQ9lldpTqjEwjaRbPrKKGkc,7089
-pyglove-0.4.5.dev202504210810.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-pyglove-0.4.5.dev202504210810.dist-info/top_level.txt,sha256=wITzJSKcj8GZUkbq-MvUQnFadkiuAv_qv5qQMw0fIow,8
-pyglove-0.4.5.dev202504210810.dist-info/RECORD,,
+pyglove-0.4.5.dev202504230810.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+pyglove-0.4.5.dev202504230810.dist-info/METADATA,sha256=R5ML7dzXaFureIQjy76mwR2AHzaVU_ZskS2X4O_uXTM,7089
+pyglove-0.4.5.dev202504230810.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+pyglove-0.4.5.dev202504230810.dist-info/top_level.txt,sha256=wITzJSKcj8GZUkbq-MvUQnFadkiuAv_qv5qQMw0fIow,8
+pyglove-0.4.5.dev202504230810.dist-info/RECORD,,