pyglove 0.4.5.dev202504040809__tar.gz → 0.5.0.dev202601100810__tar.gz

{pyglove-0.4.5.dev202504040809 → pyglove-0.5.0.dev202601100810}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyglove
-Version: 0.4.5.dev202504040809
+Version: 0.5.0.dev202601100810
 Summary: PyGlove: A library for manipulating Python objects.
 Home-page: https://github.com/google/pyglove
 Author: PyGlove Authors
@@ -23,6 +23,12 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: docstring-parser>=0.12
 Requires-Dist: termcolor>=1.1.0
+Provides-Extra: all
+Requires-Dist: docstring-parser>=0.12; extra == "all"
+Requires-Dist: termcolor>=1.1.0; extra == "all"
+Requires-Dist: fsspec>=2023.3.0; extra == "all"
+Provides-Extra: io
+Requires-Dist: fsspec>=2023.3.0; extra == "io"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier
@@ -32,6 +38,7 @@ Dynamic: home-page
 Dynamic: keywords
 Dynamic: license
 Dynamic: license-file
+Dynamic: provides-extra
 Dynamic: requires-dist
 Dynamic: summary
 

{pyglove-0.4.5.dev202504040809 → pyglove-0.5.0.dev202601100810}/pyglove/__init__.py

@@ -37,4 +37,4 @@ from pyglove.ext import *
 # pylint: enable=unused-import
 # pylint: enable=g-bad-import-order
 
-__version__ = "0.4.5"
+__version__ = "0.5.0"

{pyglove-0.4.5.dev202504040809 → pyglove-0.5.0.dev202601100810}/pyglove/core/__init__.py

@@ -1,4 +1,4 @@
-# Copyright 2019 The PyGlove Authors
+# Copyright 2025 The PyGlove Authors
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -91,6 +91,9 @@ InferredValue = symbolic.InferredValue
 Ref = symbolic.Ref
 maybe_ref = symbolic.maybe_ref
 
+# Symbolic error info
+ErrorInfo = symbolic.ErrorInfo
+
 # Decorator for declaring symbolic. members for `pg.Object`.
 members = symbolic.members
 
@@ -303,6 +306,7 @@ format = utils.format  # pylint: disable=redefined-builtin
 print = utils.print  # pylint: disable=redefined-builtin
 docstr = utils.docstr
 catch_errors = utils.catch_errors
+match_error = utils.match_error
 timeit = utils.timeit
 
 contextual_override = utils.contextual_override
@@ -349,6 +353,12 @@ from pyglove.core import coding
 
 from pyglove.core import logging
 
+#
+# Symbols from `monitoring.py`.
+#
+
+from pyglove.core import monitoring
+
 # pylint: enable=g-import-not-at-top
 # pylint: enable=reimported
 # pylint: enable=unused-import

{pyglove-0.4.5.dev202504040809 → pyglove-0.5.0.dev202601100810}/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-0.4.5.dev202504040809 → pyglove-0.5.0.dev202601100810}/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
@@ -198,8 +198,9 @@ def sandbox_call(
     except Exception as e:  # pylint: disable=broad-exception-caught
       q.put(e)
 
-  q = multiprocessing.Queue()
-  p = multiprocessing.Process(
+  ctx = multiprocessing.get_context('fork')
+  q = ctx.Queue()
+  p = ctx.Process(
       target=_call, args=tuple([q] + list(args)), kwargs=kwargs
   )
   try:

{pyglove-0.4.5.dev202504040809 → pyglove-0.5.0.dev202601100810}/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-0.4.5.dev202504040809 → pyglove-0.5.0.dev202601100810}/pyglove/core/coding/permissions.py

@@ -20,7 +20,23 @@ from typing import Optional
 from pyglove.core import utils
 
 
-class CodePermission(enum.Flag):
+class CodePermissionMeta(enum.EnumMeta):
+
+  @property
+  def BASIC(cls) -> 'CodePermission':  # pylint: disable=invalid-name
+    """Returns basic permissions."""
+    return cls.ASSIGN | cls.CALL    # pytype: disable=attribute-error
+
+  @property
+  def ALL(cls) -> 'CodePermission':  # pylint: disable=invalid-name
+    """Returns all permissions."""
+    return (
+        cls.BASIC | cls.CONDITION | cls.LOOP | cls.EXCEPTION |         # pytype: disable=attribute-error
+        cls.CLASS_DEFINITION | cls.FUNCTION_DEFINITION | cls.IMPORT    # pytype: disable=attribute-error
+    )
+
+
+class CodePermission(enum.Flag, metaclass=CodePermissionMeta):
   """Permissions for code execution."""
 
   # Allows assignment.
@@ -47,21 +63,6 @@ class CodePermission(enum.Flag):
   # Allows import.
   IMPORT = enum.auto()
 
-  @classmethod
-  @property
-  def BASIC(cls) -> 'CodePermission':    # pylint: disable=invalid-name
-    """Returns basic permissions."""
-    return CodePermission.ASSIGN | CodePermission.CALL
-
-  @classmethod
-  @property
-  def ALL(cls) -> 'CodePermission':    # pylint: disable=invalid-name
-    """Returns all permissions."""
-    return (
-        CodePermission.BASIC | CodePermission.CONDITION | CodePermission.LOOP |
-        CodePermission.EXCEPTION | CodePermission.CLASS_DEFINITION |
-        CodePermission.FUNCTION_DEFINITION | CodePermission.IMPORT)
-
 
 _TLS_CODE_RUN_PERMISSION = '__code_run_permission__'
 

{pyglove-0.4.5.dev202504040809 → pyglove-0.5.0.dev202601100810}/pyglove/core/geno/base.py

@@ -1447,6 +1447,7 @@ class DNA(symbolic.Object):
       *,
       allow_partial: bool = False,
       root_path: Optional[utils.KeyPath] = None,
+      **kwargs,
   ) -> 'DNA':
     """Class method that load a DNA from a JSON value.
 
@@ -1454,6 +1455,7 @@ class DNA(symbolic.Object):
       json_value: Input JSON value, only JSON dict is acceptable.
       allow_partial: Whether to allow elements of the list to be partial.
       root_path: KeyPath of loaded object in its object tree.
+      **kwargs: Keyword arguments that will be passed to symbolic.from_json.
 
     Returns:
       A DNA object.
@@ -1463,16 +1465,18 @@ class DNA(symbolic.Object):
       # NOTE(daiyip): DNA.parse will validate the input. Therefore, we can
       # disable runtime type check during constructing the DNA objects.
       with symbolic.enable_type_check(False):
-        dna = DNA.parse(symbolic.from_json(json_value.get('value')))
+        dna = DNA.parse(symbolic.from_json(json_value.get('value'), **kwargs))
         if 'metadata' in json_value:
           dna.rebind(
-              metadata=symbolic.from_json(json_value.get('metadata')),
+              metadata=symbolic.from_json(json_value.get('metadata'), **kwargs),
               raise_on_no_change=False, skip_notification=True)
     else:
       dna = super(DNA, cls).from_json(
           json_value,
           allow_partial=allow_partial,
-          root_path=root_path)  # pytype: disable=bad-return-type
+          root_path=root_path,
+          **kwargs,
+      )  # pytype: disable=bad-return-type
       assert isinstance(dna, DNA)
     if cloneable_metadata_keys:
       dna._cloneable_metadata_keys = set(cloneable_metadata_keys)  # pylint: disable=protected-access