xoscar 0.5.0__cp310-cp310-win_amd64.whl → 0.6.1__cp310-cp310-win_amd64.whl

xoscar/backends/pool.py

@@ -22,6 +22,7 @@ import itertools
 import logging
 import multiprocessing
 import os
+import sys
 import threading
 import traceback
 from abc import ABC, ABCMeta, abstractmethod
@@ -824,6 +825,9 @@ class ActorPoolBase(AbstractActorPool, metaclass=ABCMeta):
         with _disable_log_temporally():
             TypeDispatcher.reload_all_lazy_handlers()
 
+        if "PYTHONPATH" in os.environ:
+            sys.path.insert(0, os.environ["PYTHONPATH"])
+
         def handle_channel(channel):
             return pool.on_new_channel(channel)
 

xoscar/virtualenv/__init__.py

@@ -0,0 +1,34 @@
+# Copyright 2022-2025 XProbe Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# 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.
+
+from __future__ import annotations
+
+from pathlib import Path
+
+from .core import VirtualEnvManager
+from .uv import UVVirtualEnvManager
+
+_name_to_managers = {"uv": UVVirtualEnvManager}
+
+
+def get_virtual_env_manager(env_name: str, env_path: str | Path) -> VirtualEnvManager:
+    try:
+        manager_cls = _name_to_managers[env_name]
+    except KeyError:
+        raise ValueError(
+            f"Unknown virtualenv manager {env_name}, available: {list(_name_to_managers)}"
+        )
+
+    path = Path(env_path)
+    return manager_cls(path)

xoscar/virtualenv/core.py

@@ -0,0 +1,48 @@
+# Copyright 2022-2025 XProbe Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# 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.
+
+from __future__ import annotations
+
+from abc import ABC, abstractmethod
+from pathlib import Path
+
+
+class VirtualEnvManager(ABC):
+    @classmethod
+    @abstractmethod
+    def is_available(cls):
+        pass
+
+    def __init__(self, env_path: Path):
+        self.env_path = env_path.resolve()
+
+    @abstractmethod
+    def create_env(self, python_path: Path | None = None) -> None:
+        pass
+
+    @abstractmethod
+    def install_packages(self, packages: list[str], **kwargs):
+        pass
+
+    @abstractmethod
+    def cancel_install(self):
+        pass
+
+    @abstractmethod
+    def get_lib_path(self) -> str:
+        pass
+
+    @abstractmethod
+    def remove_env(self):
+        pass

xoscar/virtualenv/uv.py

@@ -0,0 +1,79 @@
+# Copyright 2022-2025 XProbe Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# 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.
+
+from __future__ import annotations
+
+import shutil
+import subprocess
+import sysconfig
+from pathlib import Path
+from typing import Optional
+
+from .core import VirtualEnvManager
+
+
+class UVVirtualEnvManager(VirtualEnvManager):
+    def __init__(self, env_path: Path):
+        super().__init__(env_path)
+        self._install_process: Optional[subprocess.Popen] = None
+
+    @classmethod
+    def is_available(cls):
+        return shutil.which("uv") is not None
+
+    def create_env(self, python_path: Path | None = None) -> None:
+        cmd = ["uv", "venv", str(self.env_path)]
+        if python_path:
+            cmd += ["--python", str(python_path)]
+        subprocess.run(cmd, check=True)
+
+    def install_packages(self, packages: list[str], **kwargs):
+        """
+        Install packages into the virtual environment using uv.
+        Supports pip-compatible kwargs: index_url, extra_index_url, find_links.
+        """
+        if not packages:
+            return
+
+        cmd = ["uv", "pip", "install", "-p", str(self.env_path)] + packages
+
+        # Handle known pip-related kwargs
+        if "index_url" in kwargs and kwargs["index_url"]:
+            cmd += ["-i", kwargs["index_url"]]
+        if "extra_index_url" in kwargs and kwargs["extra_index_url"]:
+            cmd += ["--extra-index-url", kwargs["extra_index_url"]]
+        if "find_links" in kwargs and kwargs["find_links"]:
+            cmd += ["-f", kwargs["find_links"]]
+        if "trusted_host" in kwargs and kwargs["trusted_host"]:
+            cmd += ["--trusted-host", kwargs["trusted_host"]]
+
+        self._install_process = process = subprocess.Popen(cmd)
+        returncode = process.wait()
+
+        self._install_process = None  # install finished, clear reference
+
+        if returncode != 0:
+            raise subprocess.CalledProcessError(returncode, cmd)
+
+    def cancel_install(self):
+        if self._install_process and self._install_process.poll() is None:
+            self._install_process.terminate()
+            self._install_process.wait()
+
+    def get_lib_path(self) -> str:
+        return sysconfig.get_path("purelib", vars={"base": str(self.env_path)})
+
+    def remove_env(self):
+        if self.env_path.exists():
+            shutil.rmtree(self.env_path, ignore_errors=True)

{xoscar-0.5.0.dist-info → xoscar-0.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xoscar
-Version: 0.5.0
+Version: 0.6.1
 Summary: Python actor framework for heterogeneous computing.
 Home-page: http://github.com/xorbitsai/xoscar
 Author: Qin Xuye
@@ -40,6 +40,7 @@ Requires-Dist: pydata-sphinx-theme>=0.3.0; extra == "dev"
 Requires-Dist: sphinx-intl>=0.9.9; extra == "dev"
 Requires-Dist: flake8>=3.8.0; extra == "dev"
 Requires-Dist: black; extra == "dev"
+Requires-Dist: uv; extra == "dev"
 Provides-Extra: doc
 Requires-Dist: ipython>=6.5.0; extra == "doc"
 Requires-Dist: sphinx; extra == "doc"

{xoscar-0.5.0.dist-info → xoscar-0.6.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 xoscar/__init__.py,sha256=dlwtB7dnDp5WME6CZVQY7d9lk1yJ9s___H5UxjGlAd4,1668
-xoscar/_utils.cp310-win_amd64.pyd,sha256=MpHLDoMyGnViWMALxdx7IVKW6CDbS7X0Vk0ue2X2-Jc,117248
+xoscar/_utils.cp310-win_amd64.pyd,sha256=Hn4TjHfIu0nDmQfMQQIAV9cQceZG2IJERtSBhFhQ_yg,117248
 xoscar/_utils.pxd,sha256=rlNbTg5lhXA-jCOLksqF4jhUlNn0xw2jx1HxdLa34pc,1193
 xoscar/_utils.pyx,sha256=18MUutw6M8Het9FcnrDMjexiJ3zxYljpTNRiYKGD-Jw,7524
 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.cp310-win_amd64.pyd,sha256=aTmgglVPNiALXKGEOBluwrz9Wm2kEHUTihWdEIxzKJs,158208
+xoscar/context.cp310-win_amd64.pyd,sha256=rHMkSsL7q4r9aBI7py6tb6IcMIMDZmMM-gLt1ZsZYSQ,158208
 xoscar/context.pxd,sha256=6n6IAbmArSRq8EjcsbS6npW8xP1jI0qOoS1fF0oyj-o,746
 xoscar/context.pyx,sha256=FOJVerGOvxe2USryXEQA0rpaFX_ScxISH6QWKUcahY8,11310
-xoscar/core.cp310-win_amd64.pyd,sha256=4-94GIBFy-6ICD_5jjjwQGLw-3yGT0lphr5y57TwP5A,336896
+xoscar/core.cp310-win_amd64.pyd,sha256=uiqIXjDDmdhq0tJt36fG9QzJNlnw4LIpQv3E6QhJEIM,336896
 xoscar/core.pxd,sha256=9IZP7dYGfnF1I-obIls8R8b6forxDOPbiM3v5nVslLk,1368
 xoscar/core.pyx,sha256=U6jCZN74MQHi7HkQRaVGm_w5q-FMsw0nnE3aU6533_Q,22756
 xoscar/debug.py,sha256=hrmxIH6zvTKasQo6PUUgXu5mgEsR0g87Fvpw7CoHipg,5257
@@ -30,9 +30,9 @@ xoscar/backends/allocate_strategy.py,sha256=DzvTlixwzTANURI2mDLHm3vcaugSPDxU6UQZ
 xoscar/backends/config.py,sha256=86j0g_Xrl8ENPzBWi296yWg9QEcljvdKK-yJbfYTvQ0,5532
 xoscar/backends/context.py,sha256=qWwksx8JxYcKR-LQA3PvXh4ReuuTTEyDaLbjpxGXcTA,16766
 xoscar/backends/core.py,sha256=fbekAxys_t1Dv7if-1R6uVvC_yAxNkXLeQ1V1ZSAfC0,11161
-xoscar/backends/message.cp310-win_amd64.pyd,sha256=kgLdppvvVYJbkkV-22Ccb91HU6-234UmTf_2beQVUcY,283648
+xoscar/backends/message.cp310-win_amd64.pyd,sha256=7lqP7pQYsRr_LmJ1Qww3jhB5W5HAGSzTJ79vFve78T4,283648
 xoscar/backends/message.pyx,sha256=lBEjMJv4VyxmeO9lHXJJazOajbFnTLak4PSBcLoPZvU,20331
-xoscar/backends/pool.py,sha256=lAYfLUqwhEvYOQWrP35dgYJf9aiT-14Leb4ohhwFrOs,62526
+xoscar/backends/pool.py,sha256=XFij5l9FCbMC1isXp4tpr6YsZVxf4GfpIRfwn-nhCGE,62638
 xoscar/backends/router.py,sha256=EjfNpQUrhFU15eYe1kRPueziHgI2gfDViUzm7ruvXDE,10817
 xoscar/backends/communication/__init__.py,sha256=i0RWfN_no-xwlzLUl8Av8zZqVWxfgBaT2qbUvOLt37s,1093
 xoscar/backends/communication/base.py,sha256=wmWTeE4lcB_ohqyJJ6MdzMGcrOqy2RSKRp8y-NDuFdY,7736
@@ -54,8 +54,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=bfXdUaj7IRYtiwsujckGJ7RqX4Kk821-1gi5kzvIOow,620544
-xoscar/collective/xoscar_pygloo.cp310-win_amd64.pyd,sha256=T_GqNw0Elpg4DO-YGCl2uhaDMm6vO_Jx0LtrXEwV76c,829952
+xoscar/collective/uv.dll,sha256=nmERAXSPqpgHqnv2_LACxMnA9QTOMUTJSUrHHxlKsdc,620544
+xoscar/collective/xoscar_pygloo.cp310-win_amd64.pyd,sha256=1depBs5XzlR0PH43dvuWLbHBUyFsHyMO4ArjYmIa2Bg,829952
 xoscar/metrics/__init__.py,sha256=RjXuuYw4I2YYgD8UY2Z5yCZk0Z56xMJ1n40O80Dtxf8,726
 xoscar/metrics/api.py,sha256=dtJ4QrIqQNXhJedeqOPs4TXKgrRGZFFN50xAd9SCfec,9144
 xoscar/metrics/backends/__init__.py,sha256=ZHepfhCDRuK9yz4pAM7bjpWDvS3Ijp1YgyynoUFLeuU,594
@@ -66,7 +66,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=tS8C49yrW_geWNEsbgW3phK1q4YN1ojI6CN-vroIFYM,876
 xoscar/serialization/aio.py,sha256=7YLXgkWpQ3ANy-TZ1qO8Mt4_J3cZFhFh2FEgUgxMT60,4873
-xoscar/serialization/core.cp310-win_amd64.pyd,sha256=F94mNi1CSkX8zvZl4law5jILyUerTRoWi9OkE1MbtEM,299520
+xoscar/serialization/core.cp310-win_amd64.pyd,sha256=ipt32mCdwaecacP0KXGWLOjua9WZtxfjmWMT9eyG1mw,299520
 xoscar/serialization/core.pxd,sha256=X-47bqBM2Kzw5SkLqICdKD0gU6CpmLsBxC3kfW--wVk,1013
 xoscar/serialization/core.pyx,sha256=ZKexLRnRwZXXn2045kR7xfM_szcoPNrDuouQCWtpFp8,30570
 xoscar/serialization/cuda.py,sha256=Fj4Cpr_YmkGceUCo0mQn8fRvmHP_5WcLdRx6epZ3RC0,3869
@@ -75,7 +75,10 @@ xoscar/serialization/mlx.py,sha256=02-GmVptgtnBr9m21BYHu3POlJXZWLqwlI44T_ZBx9o,1
 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.5.0.dist-info/METADATA,sha256=yN7Wis0PRpzNbPOj5w08bAqoOXRyaSQPQ2ndWuOomOY,9265
-xoscar-0.5.0.dist-info/WHEEL,sha256=NVXpD7b4Gxps0cd2ds5rr5TG8W4ApEwx_i5J99qMZ5E,102
-xoscar-0.5.0.dist-info/top_level.txt,sha256=vYlqqY4Nys8Thm1hePIuUv8eQePdULVWMmt7lXtX_ZA,21
-xoscar-0.5.0.dist-info/RECORD,,
+xoscar/virtualenv/__init__.py,sha256=rhJ7I6x7aXjKOCzSqsKLwqFJMh4YC2sqchEIJNEfI58,1151
+xoscar/virtualenv/core.py,sha256=pk6Pcv2Aewk1d-oTau1wyP80p8McIi7DodWL9PcDg1Q,1311
+xoscar/virtualenv/uv.py,sha256=xNNjv8Ywkwx6I1B_yQr8nkEUgJbIFNaseUCzWIo41nY,2921
+xoscar-0.6.1.dist-info/METADATA,sha256=7Qhpj9UNCrGu-3Sgfg2VW9QLaJNr5R_bzZDnlNKkkBE,9300
+xoscar-0.6.1.dist-info/WHEEL,sha256=NVXpD7b4Gxps0cd2ds5rr5TG8W4ApEwx_i5J99qMZ5E,102
+xoscar-0.6.1.dist-info/top_level.txt,sha256=vYlqqY4Nys8Thm1hePIuUv8eQePdULVWMmt7lXtX_ZA,21
+xoscar-0.6.1.dist-info/RECORD,,