xoscar 0.7.12__cp310-cp310-macosx_10_9_x86_64.whl → 0.7.14__cp310-cp310-macosx_10_9_x86_64.whl

xoscar/virtualenv/core.py

@@ -29,7 +29,13 @@ class VirtualEnvManager(ABC):
         self.env_path = env_path.resolve()
 
     @abstractmethod
-    def create_env(self, python_path: Path | None = None) -> None:
+    def exists_env(self) -> bool:
+        pass
+
+    @abstractmethod
+    def create_env(
+        self, python_path: Path | None = None, exists: str = "ignore"
+    ) -> None:
         pass
 
     @abstractmethod

xoscar/virtualenv/uv.py

@@ -65,9 +65,44 @@ class UVVirtualEnvManager(VirtualEnvManager):
                 uv_path = "uv"
         return uv_path
 
-    def create_env(self, python_path: Path | None = None) -> None:
+    def exists_env(self) -> bool:
+        """Check if virtual environment already exists."""
+        return self.env_path.exists() and (self.env_path / "pyvenv.cfg").exists()
+
+    def create_env(
+        self, python_path: Path | None = None, exists: str = "ignore"
+    ) -> None:
+        """
+        Create virtual environment.
+
+        Args:
+            python_path: Path to Python interpreter to use
+            exists: How to handle existing environment:
+                - "ignore": Skip creation if environment already exists (default)
+                - "error": Raise error if environment exists
+                - "clear": Remove existing environment and create new one
+        """
+        if self.exists_env():
+            if exists == "error":
+                raise FileExistsError(
+                    f"Virtual environment already exists at {self.env_path}"
+                )
+            elif exists == "ignore":
+                logger.info(
+                    f"Virtual environment already exists at {self.env_path}, skipping creation"
+                )
+                return
+            elif exists == "clear":
+                logger.info(f"Removing existing virtual environment at {self.env_path}")
+                self.remove_env()
+            else:
+                raise ValueError(
+                    f"Invalid exists option: {exists}. Must be one of: error, clear, ignore"
+                )
+
         uv_path = self._get_uv_path()
         cmd = [uv_path, "venv", str(self.env_path), "--system-site-packages"]
+
         if python_path:
             cmd += ["--python", str(python_path)]
         elif _is_in_pyinstaller():
@@ -115,16 +150,25 @@ class UVVirtualEnvManager(VirtualEnvManager):
     @staticmethod
     def _split_specs(
         specs: list[str], installed: dict[str, str]
-    ) -> tuple[list[str], dict[str, str]]:
+    ) -> tuple[list[str], list[str], dict[str, str]]:
         """
         Split the given requirement specs into:
+        - keep： specs that need to be kept, e.g. git+github://xxx
         - to_resolve: specs that need to be passed to the resolver (unsatisfied ones)
         - pinned: already satisfied specs, used for constraint to lock their versions
         """
+        keep: list[str] = []
         to_resolve: list[str] = []
         pinned: dict[str, str] = {}
 
         for spec_str in specs:
+            # skip git+xxx
+            if spec_str.startswith(
+                ("git+", "http://", "https://", "svn+", "hg+", "bzr+")
+            ):
+                keep.append(spec_str)
+                continue
+
             req = Requirement(spec_str)
             name = req.name.lower()
             cur_ver = installed.get(name)
@@ -150,7 +194,7 @@ class UVVirtualEnvManager(VirtualEnvManager):
                 # Parsing error, be conservative and resolve it
                 to_resolve.append(spec_str)
 
-        return to_resolve, pinned
+        return keep, to_resolve, pinned
 
     def _filter_packages_not_installed(self, packages: list[str]) -> list[str]:
         """
@@ -165,18 +209,21 @@ class UVVirtualEnvManager(VirtualEnvManager):
         }
 
         # exclude those packages that satisfied in system site packages
-        to_resolve, pinned = self._split_specs(packages, installed)
-        if not to_resolve:
+        keep, to_resolve, pinned = self._split_specs(packages, installed)
+        if not keep and not to_resolve:
             logger.debug("All requirement specifiers satisfied by system packages.")
             return []
 
-        resolved = self._resolve_install_plan(to_resolve, pinned)
-        logger.debug(f"Resolved install list: {resolved}")
-        if not resolved:
-            # no packages to install
-            return []
+        if to_resolve:
+            resolved = self._resolve_install_plan(to_resolve, pinned)
+            logger.debug(f"Resolved install list: {resolved}")
+            if not keep and not resolved:
+                # no packages to install
+                return []
+        else:
+            resolved = []
 
-        final = []
+        final = keep.copy()
         for item in resolved:
             name, version = item.split("==")
             key = name.lower()

{xoscar-0.7.12.dist-info → xoscar-0.7.14.dist-info}/METADATA

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

{xoscar-0.7.12.dist-info → xoscar-0.7.14.dist-info}/RECORD

@@ -1,7 +1,7 @@
-xoscar-0.7.12.dist-info/RECORD,,
-xoscar-0.7.12.dist-info/WHEEL,sha256=G2_osVyqlO3rQICDNwzuQ-zsxS79EB0lsq9jZ8r7Rkc,138
-xoscar-0.7.12.dist-info/top_level.txt,sha256=vYlqqY4Nys8Thm1hePIuUv8eQePdULVWMmt7lXtX_ZA,21
-xoscar-0.7.12.dist-info/METADATA,sha256=fNJZZhysdC6gdYZ67SfKjoIlNsse08kV5978cNQmnyo,9135
+xoscar-0.7.14.dist-info/RECORD,,
+xoscar-0.7.14.dist-info/WHEEL,sha256=G2_osVyqlO3rQICDNwzuQ-zsxS79EB0lsq9jZ8r7Rkc,138
+xoscar-0.7.14.dist-info/top_level.txt,sha256=vYlqqY4Nys8Thm1hePIuUv8eQePdULVWMmt7lXtX_ZA,21
+xoscar-0.7.14.dist-info/METADATA,sha256=bTouvWumRWQDRY1ApYh9XVZ7HthHSUa8ctv-___IW6U,9135
 xoscar/_utils.pyx,sha256=frgVQ5xGp92jBKc4PsPmjOlVsXlKeHWtTOAMfHmBaII,7380
 xoscar/backend.py,sha256=is436OPkZfSpQXaoqTRVta5eoye_pp45RFgCstAk2hU,1850
 xoscar/core.pxd,sha256=I_C2ka7XryyGnnAVXUVm8xfS1gtIrCs6X-9rswgOcUU,1317
@@ -10,7 +10,7 @@ xoscar/context.pxd,sha256=qKa0OyDPZtVymftSh447m-RzFZgmz8rGqQBa7qlauvc,725
 xoscar/batch.py,sha256=DpArS0L3WYJ_HVPG-6hSYEwoAFY1mY2-mlC4Jp5M_Dw,7872
 xoscar/nvutils.py,sha256=qmW4mKLU0WB2yCs198ccQOgLL02zB7Fsa-AotO3NOmg,20412
 xoscar/constants.py,sha256=QHHSREw6uWBBjQDCFqlNfTvBZgniJPGy42KSIsR8Fqw,787
-xoscar/_utils.cpython-310-darwin.so,sha256=4WfZUeB0HhwlGpHylfRyCc2RUAKz0_BJ-0A9uP_kHF0,162344
+xoscar/_utils.cpython-310-darwin.so,sha256=l_gQ96-oT60djuthhgWL9gYb4lkv5sNjhEJEsLv9qmw,162344
 xoscar/__init__.py,sha256=sy7Wtn2EuQZI0I4Az_MfsBVZm4G0DRj46qRyExgmnJk,1622
 xoscar/api.py,sha256=zxNqOjGiTIKuAip9WJ0LOoM7yevD6P5rb-sLynpZ2Zo,14648
 xoscar/utils.py,sha256=MaKiW4Vphwhh8c0yoqN8G8hbJr1zXgpf49EdvmGc1ZU,16500
@@ -20,8 +20,8 @@ xoscar/context.pyx,sha256=8CdgPnWcE9eOp3N600WgDQ03MCi8P73eUOGcfV7Zksg,10942
 xoscar/errors.py,sha256=wBlQOKsXf0Fc4skN39tDie0YZT-VIAuLNRgoDl2pZcA,1241
 xoscar/core.pyx,sha256=phN-yYV0A0QI8WFi2jCu0nc4CnShTepfDi0V7ZrLYPY,22092
 xoscar/driver.py,sha256=498fowtJr6b3FE8FIOA_Tc1Vwx88nfZw7p0FxrML0h4,1372
-xoscar/context.cpython-310-darwin.so,sha256=-B53x_dZb102buvIkfeQuFb0WAUYKOCQaCY-yz5odHc,200160
-xoscar/core.cpython-310-darwin.so,sha256=BjFsSzUnrUWArvkOeKU3v7jVTfSHDzW7DOppz31E3E0,413432
+xoscar/context.cpython-310-darwin.so,sha256=UyT-fpLpz3y1gbjJHBeYQjvxodf8tDAw4ZFfCUKR16Y,200160
+xoscar/core.cpython-310-darwin.so,sha256=ePtAilznlC7Aj2hpeeVMACcjxWUyAfKmCHxcrd31Oh4,413432
 xoscar/profiling.py,sha256=BC5OF0HzSaXv8V7w-y-B8r5gV5DgxHFoTEIF6jCMioQ,8015
 xoscar/_utils.pxd,sha256=5KYAL3jfPdejsHnrGGT2s--ZUX5SXznQWpHVSno429k,1157
 xoscar/metrics/__init__.py,sha256=9Badi7rxYikGm2dQiNCrj9GgMRBxwuR3JaEKcFZmfak,705
@@ -47,11 +47,11 @@ xoscar/serialization/cuda.py,sha256=iFUEnN4SiquBIhyieyOrfw3TnKnW-tU_vYgqOxO_DrA,
 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/core.cpython-310-darwin.so,sha256=1aggZeSSZbnPRKXA80v-tdyryxVZ7lSPkj88BN4qOBA,391832
+xoscar/serialization/core.cpython-310-darwin.so,sha256=wZRkXV0dnX6ChVZkB0iI6BbbULe-Q8zE8Jnm-S1fpKc,391832
 xoscar/serialization/mlx.py,sha256=tRu_7o6RizdRhbr88EasHrZtShimAsLy3pIEO-by29o,2118
 xoscar/backends/config.py,sha256=4tZMiXAMMS8qQ4SX_LjONLtSQVfZTx3m-IK3EqbkYdk,5375
 xoscar/backends/allocate_strategy.py,sha256=tC1Nbq2tJohahUwd-zoRYHEDX65wyuX8tmeY45uWj_w,4845
-xoscar/backends/message.cpython-310-darwin.so,sha256=vJzr34xkT92ckYFSV6rP7rxtRx4TTD_PnvXyHcho6BQ,355504
+xoscar/backends/message.cpython-310-darwin.so,sha256=-6u1tQLT6yxyPTVyTY__T5qEEvtEa-bm5owInEJeDIM,355504
 xoscar/backends/__init__.py,sha256=VHEBQcUWM5bj027W8EUf9PiJUAP7JoMrRw3Tsvy5ySw,643
 xoscar/backends/core.py,sha256=EH-fHlV9x3bnruEHaUtGYO7osKLfLJ4AQHtuzA_mr2g,10857
 xoscar/backends/context.py,sha256=XfDPG2eDhAhE6hWBEkEsHTnyyOYN9R3houlMjAL7BFw,16329
@@ -82,6 +82,6 @@ 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=qHKqI6R92SN0_OWVAX1xy9wJiZr-r0VfLHTskZtZE4U,2851
+xoscar/virtualenv/core.py,sha256=MV1lbkixGl45REHS04uaPuqPKxyVbBmVnB_umuR_eWw,2957
 xoscar/virtualenv/utils.py,sha256=mL_uATHhj82xec0-0IZ6N8yI-laPAB4t8G3alPUGtPA,2439
-xoscar/virtualenv/uv.py,sha256=PWq5Ht-uFOrwyKGF0bGZzZiuJUvGmr_OZADzRSfLKfk,9322
+xoscar/virtualenv/uv.py,sha256=8fsZtUfMfSI1FZL7eLMfF-znG7fGVwshdERiHqSvems,11119