pyglove 0.4.5.dev202501230809__py3-none-any.whl → 0.4.5.dev202501240809__py3-none-any.whl

pyglove/core/coding/execution.py

@@ -18,6 +18,7 @@ import contextlib
 import io
 import multiprocessing
 import pickle
+import queue
 from typing import Any, Callable, Dict, Optional, Union
 
 from pyglove.core import utils
@@ -197,28 +198,30 @@ def sandbox_call(
       q.put(e)
 
   q = multiprocessing.Queue()
+  p = multiprocessing.Process(
+      target=_call, args=tuple([q] + list(args)), kwargs=kwargs
+  )
   try:
-    p = multiprocessing.Process(
-        target=_call, args=tuple([q] + list(args)), kwargs=kwargs)
     p.start()
-    p.join(timeout=timeout)
+    x = q.get(timeout=timeout)
+  except queue.Empty as e:
     if p.is_alive():
       # We use `kill` instead of `terminate` to release process resources
       # right away.
       p.kill()
-      raise TimeoutError(f'Execution time exceed {timeout} seconds.')
-    x = q.get()
-    if isinstance(x, Exception):
-      raise x
-    try:
-      return pickle.loads(x)
-    except Exception as e:
-      raise errors.SerializationError(
-          'Cannot deserialize the output from sandbox.', e
-      ) from e
+    raise TimeoutError(f'Execution time exceed {timeout} seconds.') from e
   finally:
     q.close()
 
+  if isinstance(x, Exception):
+    raise x
+  try:
+    return pickle.loads(x)
+  except Exception as e:
+    raise errors.SerializationError(
+        'Cannot deserialize the output from sandbox.', e
+    ) from e
+
 
 def maybe_sandbox_call(
     func: Callable[..., Any],

{pyglove-0.4.5.dev202501230809.dist-info → pyglove-0.4.5.dev202501240809.dist-info}/METADATA

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

{pyglove-0.4.5.dev202501230809.dist-info → pyglove-0.4.5.dev202501240809.dist-info}/RECORD

@@ -5,7 +5,7 @@ pyglove/core/logging_test.py,sha256=3z_c6wnxbqDbwUmSOAZzeDPXvzoweYL5QHUQVMJ5Xgo,
 pyglove/core/coding/__init__.py,sha256=tuHIg19ZchtkOQbdFVTVLkUpBa5f1eo66XtnKw3lcIU,1645
 pyglove/core/coding/errors.py,sha256=vstDb52lk5uJef2k_0_1dqvGv8eDYNq37IrqjGJ-cNc,3302
 pyglove/core/coding/errors_test.py,sha256=fwOR8vLiRvLadubsccyE19hLHj-kThlCQt88qmUYk9M,2311
-pyglove/core/coding/execution.py,sha256=IdfAPlC7u5UwqEmsiLXowXokgm6NTvlUhzsz3h7pf3g,10867
+pyglove/core/coding/execution.py,sha256=Rq-rFPkecd70-wLnK-51L_rK01nd3J5SKfQidg3_qvY,10883
 pyglove/core/coding/execution_test.py,sha256=vTMLOCs7ceVi-XEz7-0CdPeoKqswBUQ26SzTqthcU4c,8397
 pyglove/core/coding/function_generation.py,sha256=2fCqcdcizVYtGYE6QGpw5m1kuH9Fp0OF4BjyJ4en6tw,1636
 pyglove/core/coding/function_generation_test.py,sha256=kbSwmZF8Vog0R0OTSpuzPblEbMLoRJ1TigeIrwDhHS8,2161
@@ -211,8 +211,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.dev202501230809.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-pyglove-0.4.5.dev202501230809.dist-info/METADATA,sha256=mqjqGyrVARcD5uQtx3KCV6IA9Etp6_kwcX89IAh_R-g,7067
-pyglove-0.4.5.dev202501230809.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-pyglove-0.4.5.dev202501230809.dist-info/top_level.txt,sha256=wITzJSKcj8GZUkbq-MvUQnFadkiuAv_qv5qQMw0fIow,8
-pyglove-0.4.5.dev202501230809.dist-info/RECORD,,
+pyglove-0.4.5.dev202501240809.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+pyglove-0.4.5.dev202501240809.dist-info/METADATA,sha256=lR9fscbdN1TlKb9vFXFRp2IyaPixC2Zl4E9iNCC87SE,7067
+pyglove-0.4.5.dev202501240809.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+pyglove-0.4.5.dev202501240809.dist-info/top_level.txt,sha256=wITzJSKcj8GZUkbq-MvUQnFadkiuAv_qv5qQMw0fIow,8
+pyglove-0.4.5.dev202501240809.dist-info/RECORD,,