xoscar 0.7.5__cp310-cp310-macosx_11_0_arm64.whl → 0.7.7__cp310-cp310-macosx_11_0_arm64.whl

xoscar/virtualenv/utils.py

@@ -0,0 +1,84 @@
+# 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.
+
+import logging
+import re
+import subprocess
+import sys
+import threading
+from contextlib import contextmanager
+from typing import BinaryIO, Callable, Iterator, List, Optional, TextIO, Union
+
+logger = logging.getLogger(__name__)
+
+ansi_escape = re.compile(r"\x1B\[[0-?]*[ -/]*[@-~]")
+
+
+def clean_ansi(text: str) -> str:
+    """Remove ANSI escape sequences from text."""
+    return ansi_escape.sub("", text)
+
+
+def stream_reader(
+    stream: BinaryIO, log_func: Callable[[str], None], output_stream: TextIO
+) -> None:
+    """
+    Read from the stream, write to logger, and also write to the terminal.
+    """
+    for line in iter(stream.readline, b""):
+        decoded = line.decode(errors="replace")
+        output_stream.write(decoded)
+        output_stream.flush()
+        log_func(clean_ansi(decoded.rstrip("\n")))
+
+
+@contextmanager
+def run_subprocess_with_logger(
+    cmd: Union[str, List[str]], cwd: Optional[str] = None, env: Optional[dict] = None
+) -> Iterator[subprocess.Popen]:
+    """
+    Run a subprocess, redirect stdout to logger.info and stderr to logger.error.
+    Returns the Popen object as a context manager.
+
+    :param cmd: Command to execute
+    :param kwargs: Additional arguments passed to subprocess.Popen
+    :yield: The subprocess.Popen object
+    """
+
+    process = subprocess.Popen(
+        cmd,
+        stdout=subprocess.PIPE,
+        stderr=subprocess.PIPE,
+        cwd=cwd,
+        env=env,
+        bufsize=1,
+    )
+
+    threads = [
+        threading.Thread(
+            target=stream_reader, args=(process.stdout, logger.info, sys.stdout)
+        ),
+        threading.Thread(
+            target=stream_reader, args=(process.stderr, logger.error, sys.stderr)
+        ),
+    ]
+    for t in threads:
+        t.start()
+
+    try:
+        yield process
+    finally:
+        process.wait()
+        for t in threads:
+            t.join()

xoscar/virtualenv/uv.py

@@ -24,6 +24,7 @@ from pathlib import Path
 from typing import Optional
 
 from .core import VirtualEnvManager
+from .utils import run_subprocess_with_logger
 
 UV_PATH = os.getenv("XOSCAR_UV_PATH")
 logger = logging.getLogger(__name__)
@@ -46,7 +47,14 @@ class UVVirtualEnvManager(VirtualEnvManager):
         return shutil.which("uv") is not None
 
     def create_env(self, python_path: Path | None = None) -> None:
-        uv_path = UV_PATH or "uv"
+        if (uv_path := UV_PATH) is None:
+            try:
+                from uv import find_uv_bin
+
+                uv_path = find_uv_bin()
+            except (ImportError, FileNotFoundError):
+                logger.warning("Fail to find uv bin, use system one")
+                uv_path = "uv"
         cmd = [uv_path, "venv", str(self.env_path), "--system-site-packages"]
         if python_path:
             cmd += ["--python", str(python_path)]
@@ -70,9 +78,17 @@ class UVVirtualEnvManager(VirtualEnvManager):
         # extend the ability of pip
         # maybe replace #system_torch# to the real version
         packages = self.process_packages(packages)
+        log = kwargs.pop("log", False)
 
         uv_path = UV_PATH or "uv"
-        cmd = [uv_path, "pip", "install", "-p", str(self.env_path)] + packages
+        cmd = [
+            uv_path,
+            "pip",
+            "install",
+            "-p",
+            str(self.env_path),
+            "--color=always",
+        ] + packages
 
         # Handle known pip-related kwargs
         if "index_url" in kwargs and kwargs["index_url"]:
@@ -92,8 +108,13 @@ class UVVirtualEnvManager(VirtualEnvManager):
                     cmd += [option, param_value]
 
         logger.info("Installing packages via command: %s", cmd)
-        self._install_process = process = subprocess.Popen(cmd)
-        returncode = process.wait()
+        if not log:
+            self._install_process = process = subprocess.Popen(cmd)
+            returncode = process.wait()
+        else:
+            with run_subprocess_with_logger(cmd) as process:
+                self._install_process = process
+            returncode = process.returncode
 
         self._install_process = None  # install finished, clear reference
 

{xoscar-0.7.5.dist-info → xoscar-0.7.7.dist-info}/METADATA

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

{xoscar-0.7.5.dist-info → xoscar-0.7.7.dist-info}/RECORD

@@ -1,7 +1,7 @@
-xoscar-0.7.5.dist-info/RECORD,,
-xoscar-0.7.5.dist-info/WHEEL,sha256=adaOEtyuP97RoRqh2LNimBtjXokZHsp60v8dUwE5uKE,137
-xoscar-0.7.5.dist-info/top_level.txt,sha256=vYlqqY4Nys8Thm1hePIuUv8eQePdULVWMmt7lXtX_ZA,21
-xoscar-0.7.5.dist-info/METADATA,sha256=cdE3xpdI75n6WsPnGYXmOkfeC4xVHaBTkcs4at4ZVVM,9134
+xoscar-0.7.7.dist-info/RECORD,,
+xoscar-0.7.7.dist-info/WHEEL,sha256=adaOEtyuP97RoRqh2LNimBtjXokZHsp60v8dUwE5uKE,137
+xoscar-0.7.7.dist-info/top_level.txt,sha256=vYlqqY4Nys8Thm1hePIuUv8eQePdULVWMmt7lXtX_ZA,21
+xoscar-0.7.7.dist-info/METADATA,sha256=xxkpDn2341sKmMMKzwq2RdvcuBA8N5hSCDwphsv5qcU,9134
 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=eBOLoPPPd6u7cv7d7NdKu53VPV3S5XsfZ8cgBtq-N7U,168944
+xoscar/_utils.cpython-310-darwin.so,sha256=7Ym8KRuv8G1gabZ3qr7R6GzUit3ibvWdTKWCio-wIg0,168944
 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=SHZB4oJ_euDas2pLwQJLXrdg6OwnYBmpukLFtXXSQr4,211488
-xoscar/core.cpython-310-darwin.so,sha256=_-d0t34lzLtJ4XpU3dZth1btJ8k3amsQYAIRnXHP33k,409624
+xoscar/context.cpython-310-darwin.so,sha256=_EOOLrvqC4td45PHei9o0riMVXxywmsKppVShiCWnPA,211488
+xoscar/core.cpython-310-darwin.so,sha256=vmfNzQhy6b9xawSaY3QSkBiALcXtL8tam32xmsLiIiU,409624
 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=-BZWD4VVVaZFox26Gi7rVM67R8PFI04hotGssm7T2yo,379912
+xoscar/serialization/core.cpython-310-darwin.so,sha256=fJDvhrQRg4N_vN133QFbdwrKLOKUDrn3PKWkQUFZpPQ,379912
 xoscar/serialization/mlx.py,sha256=N_cvbTUBKc14XWYsPIMz4kDstyRN1DNhb4BVRgnQm8Y,1872
 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=BLdsZyrStlVskrsTFcaUTxO7BvDRWcyy5zVrEt-o8xE,348688
+xoscar/backends/message.cpython-310-darwin.so,sha256=9QaApURszJ0c5lFR2CfCNSrvRtCj50cz-GA7MegwzeU,348688
 xoscar/backends/__init__.py,sha256=VHEBQcUWM5bj027W8EUf9PiJUAP7JoMrRw3Tsvy5ySw,643
 xoscar/backends/core.py,sha256=EH-fHlV9x3bnruEHaUtGYO7osKLfLJ4AQHtuzA_mr2g,10857
 xoscar/backends/context.py,sha256=XfDPG2eDhAhE6hWBEkEsHTnyyOYN9R3houlMjAL7BFw,16329
@@ -83,4 +83,5 @@ xoscar/aio/parallelism.py,sha256=VSsjk8wP-Bw7tLeUsTyLVNgp91thjxEfE3pCrw_vF5Q,129
 xoscar/aio/base.py,sha256=9j0f1piwfE5R5GIvV212vSD03ixdaeSzSSsO2kxJZVE,2249
 xoscar/virtualenv/__init__.py,sha256=65t9_X1DvbanNjFy366SiiWZrRTpa9SXWMXPmqayE-4,1117
 xoscar/virtualenv/core.py,sha256=dZqwg2IzHsLEERvohZx0rvBINopMRUImqxG3HHGO0q4,2744
-xoscar/virtualenv/uv.py,sha256=mR4gevfB8eFMmSLfvOYM-QhyqWyDsqO-V4bkRuhOOWA,4145
+xoscar/virtualenv/utils.py,sha256=mL_uATHhj82xec0-0IZ6N8yI-laPAB4t8G3alPUGtPA,2439
+xoscar/virtualenv/uv.py,sha256=XCjeCEntMhA0kyZS0qG7qu9AWzPtIwdeTJRLYRIOiDI,4788