xoscar 0.5.0__cp39-cp39-macosx_10_9_x86_64.whl → 0.6.1__cp39-cp39-macosx_10_9_x86_64.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,7 +1,7 @@
-xoscar-0.5.0.dist-info/RECORD,,
-xoscar-0.5.0.dist-info/WHEEL,sha256=FoxzhRRMXcm8mSmqIxEPOmfhajsNIlMLdYtHdyz-ZKQ,109
-xoscar-0.5.0.dist-info/top_level.txt,sha256=vYlqqY4Nys8Thm1hePIuUv8eQePdULVWMmt7lXtX_ZA,21
-xoscar-0.5.0.dist-info/METADATA,sha256=O_aAm6bQaZF9HvTXJsquvDrePWla--QdgpuPPnlqFds,9042
+xoscar-0.6.1.dist-info/RECORD,,
+xoscar-0.6.1.dist-info/WHEEL,sha256=FoxzhRRMXcm8mSmqIxEPOmfhajsNIlMLdYtHdyz-ZKQ,109
+xoscar-0.6.1.dist-info/top_level.txt,sha256=vYlqqY4Nys8Thm1hePIuUv8eQePdULVWMmt7lXtX_ZA,21
+xoscar-0.6.1.dist-info/METADATA,sha256=G1ypXuZl-ZjV48coc-vldRpkEeOq43k5BXIBowvxmo0,9076
 xoscar/_utils.pyx,sha256=6iqO4eTwEI-n9i39n_TKz7MWqbytMRnVNubdJ5egL6o,7279
 xoscar/backend.py,sha256=is436OPkZfSpQXaoqTRVta5eoye_pp45RFgCstAk2hU,1850
 xoscar/core.pxd,sha256=I_C2ka7XryyGnnAVXUVm8xfS1gtIrCs6X-9rswgOcUU,1317
@@ -12,18 +12,18 @@ xoscar/nvutils.py,sha256=qmW4mKLU0WB2yCs198ccQOgLL02zB7Fsa-AotO3NOmg,20412
 xoscar/constants.py,sha256=QHHSREw6uWBBjQDCFqlNfTvBZgniJPGy42KSIsR8Fqw,787
 xoscar/__init__.py,sha256=0zX8kKaio3ZIrlzB79WybcravMJw1OxPWjDspTgJFyQ,1608
 xoscar/api.py,sha256=3hztPoOxg8A_mlhWyWgVP7FMXG0PATA1TP4Rbaj7A-g,13327
-xoscar/core.cpython-39-darwin.so,sha256=INsThHiTYThtL2sNVg7reDo-i0x4Kryxz6t6KIkZBDU,448640
+xoscar/core.cpython-39-darwin.so,sha256=VocFQ0Q-nZoKXOqrA_f2szSe2AUYkInd3TVRLZwXYos,448640
 xoscar/utils.py,sha256=jUw6OICZUPBbmS1b3GE4vLctJf6fCKXrYtLtBuK-Oqc,16483
 xoscar/debug.py,sha256=9Z8SgE2WaKYQcyDo-5-DxEJQ533v7kWjrvCd28pSx3E,5069
 xoscar/libcpp.pxd,sha256=DJqBxLFOKL4iRr9Kale5UH3rbvPRD1x5bTSOPHFpz9I,1147
 xoscar/context.pyx,sha256=8CdgPnWcE9eOp3N600WgDQ03MCi8P73eUOGcfV7Zksg,10942
 xoscar/errors.py,sha256=wBlQOKsXf0Fc4skN39tDie0YZT-VIAuLNRgoDl2pZcA,1241
-xoscar/_utils.cpython-39-darwin.so,sha256=Hbzr7kqgG9qv_mzPChsKy_w-lErCREJD8O0GYsLAxo0,160888
+xoscar/_utils.cpython-39-darwin.so,sha256=n2MQVT9wHWvub1jwlSe9RvjnqV6a8AsogpB9OzcLWAs,160888
 xoscar/core.pyx,sha256=phN-yYV0A0QI8WFi2jCu0nc4CnShTepfDi0V7ZrLYPY,22092
 xoscar/driver.py,sha256=498fowtJr6b3FE8FIOA_Tc1Vwx88nfZw7p0FxrML0h4,1372
 xoscar/profiling.py,sha256=BC5OF0HzSaXv8V7w-y-B8r5gV5DgxHFoTEIF6jCMioQ,8015
 xoscar/_utils.pxd,sha256=5KYAL3jfPdejsHnrGGT2s--ZUX5SXznQWpHVSno429k,1157
-xoscar/context.cpython-39-darwin.so,sha256=Im1NDWguFiho6g2k8vtBdbzAxxfNldzLSDKzyDxycPI,200448
+xoscar/context.cpython-39-darwin.so,sha256=OHAPnNPvsMLEKG7W2rd2yuVX59CcQJTvQdkIYbYZUvI,200448
 xoscar/metrics/__init__.py,sha256=9Badi7rxYikGm2dQiNCrj9GgMRBxwuR3JaEKcFZmfak,705
 xoscar/metrics/api.py,sha256=BBlMIFvVAGVfrtpeJ1YlH9Tqhy9OzGavwvGyeHcQ0Tk,8856
 xoscar/metrics/backends/__init__.py,sha256=h_JgzSqV5lP6vQ6XX_17kE4IY4BRnvKta_7VLQAL1ms,581
@@ -43,13 +43,13 @@ xoscar/serialization/pyfury.py,sha256=sifOnVMYoS82PzZEkzkfxesmMHei23k5UAUUKUyoOY
 xoscar/serialization/core.pxd,sha256=k4RoJgX5E5LGs4jdCQ7vvcn26MabXbrWoWhkO49X6YI,985
 xoscar/serialization/__init__.py,sha256=v76XC2OQLp-Yk4_U3_IVguEylMeyRw1UrkU_DPDMh0U,856
 xoscar/serialization/numpy.py,sha256=5Kem87CvpJmzUMp3QHk4WeHU30FoQWTJJP2SwIcaQG0,2919
-xoscar/serialization/core.cpython-39-darwin.so,sha256=dSqxLhQF6E78lAhVrePrbtib1AJ84R7V84UCivvupDg,408080
+xoscar/serialization/core.cpython-39-darwin.so,sha256=muKF3dj1U5WdUtTrkv8DGka1q9-SpB3e2iIMZzWTWd8,408080
 xoscar/serialization/cuda.py,sha256=iFUEnN4SiquBIhyieyOrfw3TnKnW-tU_vYgqOxO_DrA,3758
 xoscar/serialization/scipy.py,sha256=yOEi0NB8cqQ6e2UnCZ1w006RsB7T725tIL-DM_hNcsU,2482
 xoscar/serialization/aio.py,sha256=5DySPgDxU43ec7_5Ct44-Oqt7YNSJBfuf8VdQgQlChA,4731
 xoscar/serialization/core.pyx,sha256=bjR-zXGm9qersk7kYPzpjpMIxDl_Auur4BCubRfKmfA,29626
 xoscar/serialization/mlx.py,sha256=N_cvbTUBKc14XWYsPIMz4kDstyRN1DNhb4BVRgnQm8Y,1872
-xoscar/backends/message.cpython-39-darwin.so,sha256=53f9prXDEmkuxNlKA97zPe9tlgGs_az0gKHKQtmRlJo,387584
+xoscar/backends/message.cpython-39-darwin.so,sha256=MkYBPjlV95GvnvSLCO_Yl3H5HO8kTscdJ2ssk048k9Y,387584
 xoscar/backends/config.py,sha256=4tZMiXAMMS8qQ4SX_LjONLtSQVfZTx3m-IK3EqbkYdk,5375
 xoscar/backends/allocate_strategy.py,sha256=tC1Nbq2tJohahUwd-zoRYHEDX65wyuX8tmeY45uWj_w,4845
 xoscar/backends/__init__.py,sha256=VHEBQcUWM5bj027W8EUf9PiJUAP7JoMrRw3Tsvy5ySw,643
@@ -57,7 +57,7 @@ xoscar/backends/core.py,sha256=EH-fHlV9x3bnruEHaUtGYO7osKLfLJ4AQHtuzA_mr2g,10857
 xoscar/backends/context.py,sha256=XfDPG2eDhAhE6hWBEkEsHTnyyOYN9R3houlMjAL7BFw,16329
 xoscar/backends/router.py,sha256=MVl5naz-FYf-Wla7XRn3kRxOpWV0SjKDsKNluifVA8M,10532
 xoscar/backends/message.pyx,sha256=krGVtZ1YDaZX8yWhaNHwZiudQooLvcGlw6x3Sq7jxjE,19685
-xoscar/backends/pool.py,sha256=omYqaPyv1WW1YrU3OSZhk3kr0JyfI7MYhBtTemVJKZA,60895
+xoscar/backends/pool.py,sha256=OezOhvvXAV3TpODhLHmJVgqCfowb3aA_fWZKPodm8bE,61003
 xoscar/backends/indigen/backend.py,sha256=znl_fZzWGEtLH8hZ9j9Kkf0fva25jEem2_KO7I1RVvc,1612
 xoscar/backends/indigen/__init__.py,sha256=tKHP5ClzedBRBpZsLRVErR3EUNbbDm4CY4u0rCFJr44,685
 xoscar/backends/indigen/driver.py,sha256=VGzkacYKykegW5qhCuhx01gdgBZEKJjNIyfNCnA6Nm8,952
@@ -78,3 +78,6 @@ xoscar/aio/file.py,sha256=PBtkLp-Q7XtYl-zk00s18TtgIrkNr60J3Itf66ctO1o,1486
 xoscar/aio/lru.py,sha256=rpXCqSLtPV5xnWtd6uDwQQFGgIPEgvmWEQDkPNUx9cM,6311
 xoscar/aio/parallelism.py,sha256=VSsjk8wP-Bw7tLeUsTyLVNgp91thjxEfE3pCrw_vF5Q,1293
 xoscar/aio/base.py,sha256=9j0f1piwfE5R5GIvV212vSD03ixdaeSzSSsO2kxJZVE,2249
+xoscar/virtualenv/__init__.py,sha256=65t9_X1DvbanNjFy366SiiWZrRTpa9SXWMXPmqayE-4,1117
+xoscar/virtualenv/core.py,sha256=YMN6yHoNeEc8ecbJMbZkKeWKUABK1mUZ_OYOjcbRqWs,1263
+xoscar/virtualenv/uv.py,sha256=l9w5JxETbW2wN8vfZR7k7aKTz4cWV4MvH06MoogBOVc,2842