xoscar 0.4.3__cp312-cp312-win_amd64.whl → 0.4.4__cp312-cp312-win_amd64.whl

xoscar/backends/core.py

@@ -208,6 +208,40 @@ class ActorCallerThreadLocal:
         _ = [task.cancel() for task in self._clients.values()]
 
 
+def _cancel_all_tasks(loop):
+    to_cancel = asyncio.all_tasks(loop)
+    if not to_cancel:
+        return
+
+    for task in to_cancel:
+        task.cancel()
+
+    loop.run_until_complete(asyncio.gather(*to_cancel, return_exceptions=True))
+
+    for task in to_cancel:
+        if task.cancelled():
+            continue
+        if task.exception() is not None:
+            loop.call_exception_handler(
+                {
+                    "message": "unhandled exception during asyncio.run() shutdown",
+                    "exception": task.exception(),
+                    "task": task,
+                }
+            )
+
+
+def _safe_run_forever(loop):
+    loop.run_forever()
+    _cancel_all_tasks(loop)
+
+
+def _safe_exit_thread(loop, thread):
+    # To avoid _enter_buffered_busy: could not acquire lock
+    loop.call_soon_threadsafe(loop.stop)
+    thread.join()
+
+
 class ActorCaller:
     __slots__ = "_thread_local"
 
@@ -215,9 +249,11 @@ class ActorCaller:
         pass
 
     _close_loop = asyncio.new_event_loop()
-    _close_thread = threading.Thread(target=_close_loop.run_forever, daemon=True)
+    _close_thread = threading.Thread(
+        target=_safe_run_forever, args=(_close_loop,), daemon=True
+    )
     _close_thread.start()
-    atexit.register(_close_loop.call_soon_threadsafe, _close_loop.stop)
+    atexit.register(_safe_exit_thread, _close_loop, _close_thread)
 
     def __init__(self):
         self._thread_local = threading.local()

{xoscar-0.4.3.dist-info → xoscar-0.4.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xoscar
-Version: 0.4.3
+Version: 0.4.4
 Summary: Python actor framework for heterogeneous computing.
 Home-page: http://github.com/xorbitsai/xoscar
 Author: Qin Xuye

{xoscar-0.4.3.dist-info → xoscar-0.4.4.dist-info}/RECORD

@@ -1,5 +1,5 @@
 xoscar/__init__.py,sha256=dlwtB7dnDp5WME6CZVQY7d9lk1yJ9s___H5UxjGlAd4,1668
-xoscar/_utils.cp312-win_amd64.pyd,sha256=bK37DveUp6F-hlp6-4oEapJMfB0cEVImzic8BMTid6s,114688
+xoscar/_utils.cp312-win_amd64.pyd,sha256=gYtFeq9ao2o9KVhVmiJiUUQGhBOMJZAq_O52rYjBx64,114688
 xoscar/_utils.pxd,sha256=rlNbTg5lhXA-jCOLksqF4jhUlNn0xw2jx1HxdLa34pc,1193
 xoscar/_utils.pyx,sha256=5Wvind3AQ3JaMK7Zv9SjhiPO6LEol2hW7_fMncn69go,7300
 xoscar/_version.py,sha256=bsfCVAo_o9LkiP3AjPsP4SRRqhjuS0t4D1WGJPzbdls,24412
@@ -7,10 +7,10 @@ xoscar/api.py,sha256=B5oXv4vgMxMteh1YNaBmNFDrUFmYa_dCdzfaWwwZnCo,13820
 xoscar/backend.py,sha256=8G5JwjoOT6Q2slb11eXNApxgcmvNQUCdQzkoIMDwLcQ,1917
 xoscar/batch.py,sha256=Jk5BSpvMFAV9DrRy0a9tgPvIo_dt8cbJReZBL0cnOPc,8128
 xoscar/constants.py,sha256=GJ1KEOxwnqksc9K_GH42TSWpQECeC6ti3KJmE3PUcTw,810
-xoscar/context.cp312-win_amd64.pyd,sha256=jiBNYUkTr3MU3CTb60T5Mb4hsuqvJBxlVfqYPeFmei8,159232
+xoscar/context.cp312-win_amd64.pyd,sha256=UuU0KCW8zoqNrO6P1eTKoEDHry4GYi_g0WGLyjD5-4w,159232
 xoscar/context.pxd,sha256=6n6IAbmArSRq8EjcsbS6npW8xP1jI0qOoS1fF0oyj-o,746
 xoscar/context.pyx,sha256=FOJVerGOvxe2USryXEQA0rpaFX_ScxISH6QWKUcahY8,11310
-xoscar/core.cp312-win_amd64.pyd,sha256=ropp29FwfP-IRfYFuA84YQVAbRR1KjV05P7FlZMYwos,329728
+xoscar/core.cp312-win_amd64.pyd,sha256=9G6fFAJlG7lW5W3k_MnpkR6YlK8DVZQ3Ji7K3JcwSDc,329728
 xoscar/core.pxd,sha256=dGv62H92PFG98SVILuF641kY-NWFEt1FYqqOX3WY5RQ,1330
 xoscar/core.pyx,sha256=0YvJW2AHgymyfsAlPGvIFw65J5gTKO3PK2p1wl4VlJ0,22388
 xoscar/debug.py,sha256=hrmxIH6zvTKasQo6PUUgXu5mgEsR0g87Fvpw7CoHipg,5257
@@ -29,8 +29,8 @@ xoscar/backends/__init__.py,sha256=g9OllTquu9MRB5nySVoyiRv2z-_OSALWrOhwt7L9WXc,6
 xoscar/backends/allocate_strategy.py,sha256=DzvTlixwzTANURI2mDLHm3vcaugSPDxU6UQZb89KH0U,5005
 xoscar/backends/config.py,sha256=7nmvU_19zYR7n-bT8BNasbjntwmobmMiM7wN7O6Lujc,5129
 xoscar/backends/context.py,sha256=NukXzBwq9ZwuiN1y6RE1hfNGsW589hDtJAVwu-DV9E0,15874
-xoscar/backends/core.py,sha256=UBgPX-9ZdC2b6Ci5dGowlqdp1Sb2JU9ULYOhHkd9wOQ,9623
-xoscar/backends/message.cp312-win_amd64.pyd,sha256=13o0lAuxceovaQhCGM6ep7d6b3UYlqgYWavOcM8ppe0,267264
+xoscar/backends/core.py,sha256=NqzvYyd6E18osehrMdDC4JWrrED4HArKhUZv0_gV3AA,10541
+xoscar/backends/message.cp312-win_amd64.pyd,sha256=85tB7APIvUKJDbBSsYW5lN9sHCxWfe-oqHHOW6bwu1w,267264
 xoscar/backends/message.pyi,sha256=JMe3_gwmBCBVJPA2TpxY0ygLyYpTsizHE9-C8syjyIM,6351
 xoscar/backends/message.pyx,sha256=23j-fjxOVONPWY0-C5sfCOMbZCx4PTxN0GU8YfrLNeg,18529
 xoscar/backends/pool.py,sha256=nh2DTrcOrZc75TG2v7MXeGLMLhEO4GsObD7zirymZ3c,61445
@@ -55,8 +55,8 @@ xoscar/collective/common.py,sha256=9c7xq3IOUvfA0I9GnpalUqXZOzmF6IEILv4zL64BYVE,3
 xoscar/collective/core.py,sha256=191aPxbUgWpjzrqyozndImDAQhZFmqoQdBkHFLDfXN0,24239
 xoscar/collective/process_group.py,sha256=kTPbrLMJSGhqbiWvTIiz-X3W0rZWd_CFn_zUIlXbOlM,23286
 xoscar/collective/utils.py,sha256=p3WEVtXvnVhkuO5mRgQBhBRFr1dKHcDKMjrbMyuiyfg,1219
-xoscar/collective/uv.dll,sha256=bk3mH0ivLhkidHz0WeuEEs7YyQoKOK8kJOY9_eWNja4,620544
-xoscar/collective/xoscar_pygloo.cp312-win_amd64.pyd,sha256=IHREkPVZsXVInchDw5iLCV_M49xbBsifpCud6fS2acw,829952
+xoscar/collective/uv.dll,sha256=HawOnCwnhhH6nLvvNgcS52AmOqAfQpe9KW2fZ4ImfCA,620544
+xoscar/collective/xoscar_pygloo.cp312-win_amd64.pyd,sha256=2H3QZGwoIoMO_hPSlEBISibB9oGB8OCITt3CBWj19pU,829952
 xoscar/collective/xoscar_pygloo.pyi,sha256=6KRzElgNBBKWh-VivUw1b5Dolp17MgwA91hQo33EysU,7616
 xoscar/metrics/__init__.py,sha256=RjXuuYw4I2YYgD8UY2Z5yCZk0Z56xMJ1n40O80Dtxf8,726
 xoscar/metrics/api.py,sha256=dtJ4QrIqQNXhJedeqOPs4TXKgrRGZFFN50xAd9SCfec,9144
@@ -68,7 +68,7 @@ xoscar/metrics/backends/prometheus/__init__.py,sha256=ZHepfhCDRuK9yz4pAM7bjpWDvS
 xoscar/metrics/backends/prometheus/prometheus_metric.py,sha256=65hb8O3tmsEJ7jgOrIwl_suj9SE5Tmqcfjuk0urkLvE,2120
 xoscar/serialization/__init__.py,sha256=NOAn046vnHEkx--82BKNinV8EpyOfT5hqfRBGnKl56s,866
 xoscar/serialization/aio.py,sha256=7YLXgkWpQ3ANy-TZ1qO8Mt4_J3cZFhFh2FEgUgxMT60,4873
-xoscar/serialization/core.cp312-win_amd64.pyd,sha256=ThprTXIp_p6_yP0wx_txUQiGnCLrSx2QtaKfKZkot38,291840
+xoscar/serialization/core.cp312-win_amd64.pyd,sha256=L59_HkQQLlgklaFUzKz-2FgDU8kDuGPyAV7aFb181XY,291840
 xoscar/serialization/core.pxd,sha256=X-47bqBM2Kzw5SkLqICdKD0gU6CpmLsBxC3kfW--wVk,1013
 xoscar/serialization/core.pyi,sha256=Zof9373qy2lmjenSuMznEiTSCNW6WQB7rmSXrRbjUo0,1931
 xoscar/serialization/core.pyx,sha256=ZKexLRnRwZXXn2045kR7xfM_szcoPNrDuouQCWtpFp8,30570
@@ -77,7 +77,7 @@ xoscar/serialization/exception.py,sha256=t6yZn_Ate04UE1RbabNh7mu739sdtwarjuPXWhA
 xoscar/serialization/numpy.py,sha256=C6WVx-Sdl2OHBAvVY34DFjAKXlekMbpc2ni6bR8wxYo,3001
 xoscar/serialization/pyfury.py,sha256=3ucal29Hr7PX9_1SfB2x43FE2xw_C0rLkVv3foL7qwM,1200
 xoscar/serialization/scipy.py,sha256=9ph-yoRoNiwUZTwQrn35U60VPirWlncXNAg6EXvqMR4,2554
-xoscar-0.4.3.dist-info/METADATA,sha256=1h2geTxffR1yRtO6UVUH9Cn4tb7vKSXfHxjGZheXgZ0,9265
-xoscar-0.4.3.dist-info/WHEEL,sha256=pWXrJbnZSH-J-PhYmKs2XNn4DHCPNBYq965vsBJBFvA,101
-xoscar-0.4.3.dist-info/top_level.txt,sha256=vYlqqY4Nys8Thm1hePIuUv8eQePdULVWMmt7lXtX_ZA,21
-xoscar-0.4.3.dist-info/RECORD,,
+xoscar-0.4.4.dist-info/METADATA,sha256=yy8pXhKcYJlGtW4UqAy9N-JrCAFgKdcnuOrN6Ev1jjo,9265
+xoscar-0.4.4.dist-info/WHEEL,sha256=pWXrJbnZSH-J-PhYmKs2XNn4DHCPNBYq965vsBJBFvA,101
+xoscar-0.4.4.dist-info/top_level.txt,sha256=vYlqqY4Nys8Thm1hePIuUv8eQePdULVWMmt7lXtX_ZA,21
+xoscar-0.4.4.dist-info/RECORD,,