pyglove 0.5.0.dev202510160810__py3-none-any.whl → 0.5.0.dev202510170811__py3-none-any.whl

pyglove/core/utils/contextual.py

@@ -15,6 +15,7 @@
 
 import contextlib
 import dataclasses
+import inspect
 import threading
 from typing import Any, Callable, ContextManager, Iterator, Optional
 
@@ -89,10 +90,14 @@ def with_contextual_override(func: Callable[..., Any]) -> Callable[..., Any]:
   with contextual_override() as current_context:
     pass
 
-  def _func(*args, **kwargs) -> Any:
-    with contextual_override(**current_context):
-      return func(*args, **kwargs)
-
+  if inspect.iscoroutinefunction(func):
+    async def _func(*args, **kwargs) -> Any:
+      with contextual_override(**current_context):
+        return await func(*args, **kwargs)
+  else:
+    def _func(*args, **kwargs) -> Any:
+      with contextual_override(**current_context):
+        return func(*args, **kwargs)
   return _func
 
 

pyglove/core/utils/contextual_test.py

@@ -11,6 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+import asyncio
 import concurrent.futures
 import unittest
 from pyglove.core.utils import contextual
@@ -83,6 +84,15 @@ class ContextualTest(unittest.TestCase):
           [3]
       )
 
+  def test_with_contextual_override_async_func(self):
+    async def func(i):
+      del i
+      return contextual.contextual_value('x')
+
+    with contextual.contextual_override(x=3):
+      self.assertEqual(
+          asyncio.run(contextual.with_contextual_override(func)(0)), 3
+      )
 
 if __name__ == '__main__':
   unittest.main()

{pyglove-0.5.0.dev202510160810.dist-info → pyglove-0.5.0.dev202510170811.dist-info}/METADATA

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

{pyglove-0.5.0.dev202510160810.dist-info → pyglove-0.5.0.dev202510170811.dist-info}/RECORD

@@ -139,8 +139,8 @@ pyglove/core/typing/value_specs_test.py,sha256=eGXVxdduIM-oEaapJS9Kh7WSQHRUFegLI
 pyglove/core/utils/__init__.py,sha256=2aw4n1kYG9xlX2tWI-H5i25cBuK1ME9Lmf-F31VlKEk,8657
 pyglove/core/utils/common_traits.py,sha256=PWxOgPhG5H60ZwfO8xNAEGRjFUqqDZQBWQYomOfvdy8,3640
 pyglove/core/utils/common_traits_test.py,sha256=DIuZB_1xfmeTVfWnGOguDQcDAM_iGgBOe8C-5CsIqBc,1122
-pyglove/core/utils/contextual.py,sha256=RxBQkDM2gB6QwZj_2oMels6oh-zQPGJJlinZbbqHuYQ,5148
-pyglove/core/utils/contextual_test.py,sha256=OOcthquVyAekTqt1RyYcEMHaockMIokpbv4pSf13Nzs,3252
+pyglove/core/utils/contextual.py,sha256=_EO_ubCcmI81QyYyyucm3QcH1asQWSeXvT2Xa4KONs8,5355
+pyglove/core/utils/contextual_test.py,sha256=g5P6IBf6XgQTUW83QTDC1K5JQJ9bxP6ULiAj2ko4mMo,3552
 pyglove/core/utils/docstr_utils.py,sha256=5BY40kXozPKVGOB0eN8jy1P5_GHIzqFJ9FXAu_kzxaw,5119
 pyglove/core/utils/docstr_utils_test.py,sha256=TBHNwvhGyyoEs7dNOv5bW7h3B_y2smDyoAR9uPDiidI,4179
 pyglove/core/utils/error_utils.py,sha256=5if1LlouHMx9EkZt-rhlkDU53Qc-hmXgjhJHhhI125s,6392
@@ -218,8 +218,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.dev202510160810.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-pyglove-0.5.0.dev202510160810.dist-info/METADATA,sha256=65NWYsHR82HetqC5evpjVKFUh_P8ARXiZkKLwu5MOss,7089
-pyglove-0.5.0.dev202510160810.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pyglove-0.5.0.dev202510160810.dist-info/top_level.txt,sha256=wITzJSKcj8GZUkbq-MvUQnFadkiuAv_qv5qQMw0fIow,8
-pyglove-0.5.0.dev202510160810.dist-info/RECORD,,
+pyglove-0.5.0.dev202510170811.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+pyglove-0.5.0.dev202510170811.dist-info/METADATA,sha256=ddAxv8wl3xERViIzNE1CjqFa52kA5OgHzQzxyVlgpy8,7089
+pyglove-0.5.0.dev202510170811.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pyglove-0.5.0.dev202510170811.dist-info/top_level.txt,sha256=wITzJSKcj8GZUkbq-MvUQnFadkiuAv_qv5qQMw0fIow,8
+pyglove-0.5.0.dev202510170811.dist-info/RECORD,,