roar-cli 0.3.0__tar.gz → 0.3.1__tar.gz

{roar_cli-0.3.0 → roar_cli-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: roar-cli
-Version: 0.3.0
+Version: 0.3.1
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
@@ -69,7 +69,7 @@ Requires Python 3.10+.
 
 > For the full prereqs, platform support matrix, tracer-backend setup,
 > macOS SIP notes, and sdist build steps, see the canonical
-> [Installing roar](https://glaas.ai/docs/roar-guide#installing-roar) docs page.
+> [Installation](https://glaas.ai/docs/installation) docs page.
 > What's below is a TL;DR.
 
 ### Platform Support

{roar_cli-0.3.0 → roar_cli-0.3.1}/README.md

@@ -20,7 +20,7 @@ Requires Python 3.10+.
 
 > For the full prereqs, platform support matrix, tracer-backend setup,
 > macOS SIP notes, and sdist build steps, see the canonical
-> [Installing roar](https://glaas.ai/docs/roar-guide#installing-roar) docs page.
+> [Installation](https://glaas.ai/docs/installation) docs page.
 > What's below is a TL;DR.
 
 ### Platform Support

{roar_cli-0.3.0 → roar_cli-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "roar-cli"
-version = "0.3.0"
+version = "0.3.1"
 description = "Reproducibility and provenance tracker for ML training pipelines"
 authors = [
     { name="TReqs Team", email="info@treqs.ai" }

{roar_cli-0.3.0 → roar_cli-0.3.1}/roar/execution/framework/runtime_imports.py

@@ -22,6 +22,17 @@ class RuntimeImportController:
     def __init__(self, environ: MutableMapping[str, str]):
         self._environ = environ
         self._initialized_backend_names: set[str] = set()
+        self._backend_dispatch_disabled = False
+
+    def disable_backend_dispatch(self) -> None:
+        """Suppress backend init / observe / patch for the lifetime of this controller.
+
+        Used when the traced Python's ABI doesn't match roar's bundled compiled
+        deps — loading a backend plugin would trigger an ``ImportError`` from
+        wrong-ABI wheels (pydantic_core, etc.). The stdlib tracker hooks
+        (file opens, environ reads, import names) keep running.
+        """
+        self._backend_dispatch_disabled = True
 
     def resolve_selected_backend(self) -> ExecutionBackend | None:
         backend_name = str(self._environ.get(ROAR_EXECUTION_BACKEND_ENV) or "").strip()
@@ -33,11 +44,16 @@ class RuntimeImportController:
             return None
 
     def initialize_selected_backend(self) -> None:
+        if self._backend_dispatch_disabled:
+            return
         backend = self.resolve_selected_backend()
         if backend is not None:
             self._initialize_backend(backend)
 
     def handle_import(self, module_name: str, module: Any) -> ExecutionBackend | None:
+        if self._backend_dispatch_disabled:
+            return None
+
         matched_backend = match_execution_backend_for_module(module_name)
         if matched_backend is not None:
             self._environ.setdefault(ROAR_EXECUTION_BACKEND_ENV, matched_backend.name)

{roar_cli-0.3.0 → roar_cli-0.3.1}/roar/execution/runtime/inject/sitecustomize.py

@@ -20,6 +20,7 @@ def _append_roar_runtime_pythonpath() -> None:
 _append_roar_runtime_pythonpath()
 
 from roar.execution.framework.runtime_imports import RuntimeImportController
+from roar.execution.runtime.inject.support import abi_minor_version, bundled_abi_tag
 from roar.execution.runtime.inject.tracker import RuntimeInjectionTracker
 
 LOG_FILE = os.environ.get("ROAR_LOG_FILE")
@@ -42,7 +43,22 @@ _runtime_tracker.install()
 
 
 if os.environ.get("ROAR_WRAP") == "1":
-    _runtime_import_controller.initialize_selected_backend()
+    _bundled_abi = abi_minor_version(bundled_abi_tag(_ROAR_INJECT_DIR))
+    _running_abi = (sys.version_info.major, sys.version_info.minor)
+    if _bundled_abi is not None and _bundled_abi != _running_abi:
+        sys.stderr.write(
+            f"roar: traced Python is {_running_abi[0]}.{_running_abi[1]} but "
+            f"roar-cli was installed under Python "
+            f"{_bundled_abi[0]}.{_bundled_abi[1]}.\n"
+            f"  Backend integrations (Ray, OSMO) are disabled for this run.\n"
+            f"  File I/O is still captured.\n"
+            f"  To re-enable backends, reinstall under the matching Python:\n"
+            f"    uv tool install --python python{_running_abi[0]}.{_running_abi[1]} "
+            f"roar-cli --force\n"
+        )
+        _runtime_import_controller.disable_backend_dispatch()
+    else:
+        _runtime_import_controller.initialize_selected_backend()
 
 
 atexit.register(_runtime_tracker.write_log)

roar_cli-0.3.1/roar/execution/runtime/inject/support.py

@@ -0,0 +1,98 @@
+from __future__ import annotations
+
+import contextlib
+import os
+import re
+import shutil
+import sys
+import threading
+
+_roar_suppress = threading.local()
+
+
+def bundled_abi_tag(inject_dir: str) -> str | None:
+    """Return the cpython ABI tag of roar's bundled compiled deps, or None.
+
+    Walks up from ``inject_dir`` to find the enclosing ``site-packages`` and
+    parses the CPython ABI tag from a known compiled dependency's ``.so``
+    filename (e.g. ``_pydantic_core.cpython-313-x86_64-linux-gnu.so`` →
+    ``"cpython-313"``). Returns ``None`` if the layout doesn't match — callers
+    should treat ``None`` as "don't gate" rather than assuming mismatch.
+    """
+    site_pkg = inject_dir
+    while os.path.basename(site_pkg) != "site-packages":
+        parent = os.path.dirname(site_pkg)
+        if parent == site_pkg:
+            return None
+        site_pkg = parent
+
+    for known_pkg in ("pydantic_core", "blake3"):
+        pkg_dir = os.path.join(site_pkg, known_pkg)
+        if not os.path.isdir(pkg_dir):
+            continue
+        try:
+            entries = os.listdir(pkg_dir)
+        except OSError:
+            continue
+        for filename in entries:
+            for token in filename.split("."):
+                if token.startswith("cpython-"):
+                    m = re.match(r"cpython-\d+", token)
+                    if m:
+                        return m.group()
+    return None
+
+
+def abi_minor_version(tag: str | None) -> tuple[int, int] | None:
+    """Extract ``(major, minor)`` from an ABI tag like ``cp313`` or ``cpython-313``."""
+    if not tag:
+        return None
+    match = re.search(r"\d+", tag)
+    if not match:
+        return None
+    digits = match.group()
+    if len(digits) < 2:
+        return None
+    return (int(digits[0]), int(digits[1:]))
+
+
+def is_suppressed() -> bool:
+    return bool(getattr(_roar_suppress, "active", False))
+
+
+class SuppressTracking:
+    """Context manager: file operations inside are not tracked."""
+
+    def __enter__(self):
+        _roar_suppress.active = True
+        return self
+
+    def __exit__(self, *_):
+        _roar_suppress.active = False
+
+
+def merge_working_dir(source_dir: str, target_dir: str) -> None:
+    for entry in os.listdir(source_dir):
+        src = os.path.join(source_dir, entry)
+        dst = os.path.join(target_dir, entry)
+        try:
+            if os.path.isdir(src):
+                shutil.copytree(src, dst, dirs_exist_ok=True)
+            else:
+                shutil.copy2(src, dst)
+        except Exception:
+            continue
+
+
+def warn_runtime(message: str, *args: object) -> None:
+    text = message % args if args else message
+    try:
+        from roar.core.logging import get_logger
+
+        get_logger().warning(text)
+        return
+    except Exception:
+        pass
+
+    with contextlib.suppress(Exception):
+        sys.stderr.write(text + "\n")

roar_cli-0.3.0/roar/execution/runtime/inject/support.py

@@ -1,51 +0,0 @@
-from __future__ import annotations
-
-import contextlib
-import os
-import shutil
-import sys
-import threading
-
-_roar_suppress = threading.local()
-
-
-def is_suppressed() -> bool:
-    return bool(getattr(_roar_suppress, "active", False))
-
-
-class SuppressTracking:
-    """Context manager: file operations inside are not tracked."""
-
-    def __enter__(self):
-        _roar_suppress.active = True
-        return self
-
-    def __exit__(self, *_):
-        _roar_suppress.active = False
-
-
-def merge_working_dir(source_dir: str, target_dir: str) -> None:
-    for entry in os.listdir(source_dir):
-        src = os.path.join(source_dir, entry)
-        dst = os.path.join(target_dir, entry)
-        try:
-            if os.path.isdir(src):
-                shutil.copytree(src, dst, dirs_exist_ok=True)
-            else:
-                shutil.copy2(src, dst)
-        except Exception:
-            continue
-
-
-def warn_runtime(message: str, *args: object) -> None:
-    text = message % args if args else message
-    try:
-        from roar.core.logging import get_logger
-
-        get_logger().warning(text)
-        return
-    except Exception:
-        pass
-
-    with contextlib.suppress(Exception):
-        sys.stderr.write(text + "\n")