openrunner-sdk 2.2.0__tar.gz → 2.4.0__tar.gz

{openrunner_sdk-2.2.0 → openrunner_sdk-2.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrunner-sdk
-Version: 2.2.0
+Version: 2.4.0
 Summary: OpenRunner SDK - W&B-compatible ML experiment tracking client
 Project-URL: Homepage, https://github.com/jqueguiner/openrunner
 Project-URL: Repository, https://github.com/jqueguiner/openrunner

{openrunner_sdk-2.2.0 → openrunner_sdk-2.4.0}/openrunner/api_client.py

@@ -490,6 +490,86 @@ class APIClient:
             logger.warning("use_artifact failed: %s", e)
             return None
 
+    def download_artifact(
+        self,
+        run_id: str,
+        artifact_name: str,
+        dest_dir: str = ".",
+        version: int | None = None,
+        alias: str | None = None,
+    ) -> str | None:
+        """Download all files from an artifact version to a local directory.
+
+        Args:
+            run_id: Run ID that used/created the artifact
+            artifact_name: Artifact name (or "name:alias")
+            dest_dir: Local directory to save files (created if needed)
+            version: Specific version number (optional)
+            alias: Alias name like "latest", "best" (optional)
+
+        Returns:
+            Path to the download directory, or None on failure.
+
+        Example:
+            path = client.download_artifact(run_id, "model-checkpoint", "./models")
+            # Files saved to ./models/model-checkpoint/v3/...
+        """
+        import os
+        from pathlib import Path
+
+        info = self.use_artifact(run_id, artifact_name, version=version, alias=alias)
+        if not info:
+            logger.warning("download_artifact: use_artifact returned no data")
+            return None
+
+        # Extract version info and files
+        ver = info.get("version", version or 1)
+        files = info.get("files", [])
+        if not files:
+            logger.warning("download_artifact: no files in artifact")
+            return None
+
+        # Create destination directory
+        name_clean = artifact_name.split(":")[0].replace("/", "_")
+        out_dir = Path(dest_dir) / name_clean / f"v{ver}"
+        out_dir.mkdir(parents=True, exist_ok=True)
+
+        downloaded = 0
+        for f in files:
+            url = f.get("presigned_url") or f.get("url") or f.get("download_url")
+            fname = f.get("name") or f.get("path") or f"file_{downloaded}"
+            if not url:
+                continue
+
+            # Try presigned URL first, fall back to proxy
+            data = self.download_file_from_presigned_url(url)
+            if data is None and "localhost" in url:
+                # Try proxy
+                key = f.get("storage_key") or f.get("key")
+                if key:
+                    try:
+                        resp = self._request("GET", f"/storage/download?key={key}")
+                        if resp.status_code == 200:
+                            data = resp.content
+                    except Exception:
+                        pass
+
+            if data:
+                file_path = out_dir / fname
+                file_path.parent.mkdir(parents=True, exist_ok=True)
+                file_path.write_bytes(data)
+                downloaded += 1
+                logger.info("downloaded: %s (%d bytes)", fname, len(data))
+            else:
+                logger.warning("failed to download: %s", fname)
+
+        if downloaded == 0:
+            logger.warning("download_artifact: no files downloaded")
+            return None
+
+        logger.info("artifact downloaded: %d files → %s", downloaded, out_dir)
+        return str(out_dir)
+
     def set_alias(
         self,
         artifact_id: str,
@@ -684,21 +764,37 @@ class APIClient:
         presigned_url: str,
         data_bytes: bytes,
         content_type: str = "image/png",
+        run_id: str = "",
+        key: str = "",
     ) -> bool:
         """PUT bytes to a presigned URL with Content-Type header.
 
-        Uses httpx directly (not self._client) because presigned URLs
-        are absolute and should not have a base_url prefix.
+        Falls back to API proxy upload if presigned URL is unreachable
+        (e.g., MinIO not exposed publicly).
         """
         try:
-            resp = httpx.put(
-                presigned_url,
-                content=data_bytes,
-                headers={"Content-Type": content_type},
-                timeout=300.0,
-            )
-            resp.raise_for_status()
-            return True
+            try:
+                resp = httpx.put(
+                    presigned_url,
+                    content=data_bytes,
+                    headers={"Content-Type": content_type},
+                    timeout=300.0,
+                )
+                resp.raise_for_status()
+                return True
+            except (httpx.ConnectError, httpx.ConnectTimeout):
+                # Presigned URL unreachable — use API proxy
+                if run_id and key:
+                    import os
+                    fname = os.path.basename(key) or key.replace("/", "_")
+                    proxy_resp = self._request(
+                        "PUT",
+                        f"/runs/{run_id}/files/media/{fname}",
+                        content=data_bytes,
+                        headers={"Content-Type": content_type},
+                    )
+                    return proxy_resp.status_code == 200
+                return False
         except Exception as e:
             logger.warning("upload_media_bytes failed: %s", e)
             return False

{openrunner_sdk-2.2.0 → openrunner_sdk-2.4.0}/openrunner/run.py

@@ -1154,18 +1154,28 @@ class Run:
         name: str,
         version: int | None = None,
         alias: str | None = None,
-    ):
-        """Convenience: download a model artifact.
+        dest_dir: str = "./artifacts",
+    ) -> str | None:
+        """Download a model artifact to disk.
 
         Args:
             name: Model artifact name (supports "name:alias" syntax).
             version: Specific version number, or None for latest.
             alias: Alias name to resolve.
+            dest_dir: Local directory for downloaded files.
 
         Returns:
             Path to the local artifact directory, or None on failure.
         """
-        return self.use_artifact(name, version=version, alias=alias)
+        if not self._client:
+            return None
+        return self._client.download_artifact(
+            run_id=self._run_id,
+            artifact_name=name,
+            dest_dir=dest_dir,
+            version=version,
+            alias=alias,
+        )
 
     def link_model(
         self,

{openrunner_sdk-2.2.0 → openrunner_sdk-2.4.0}/openrunner/sender.py

@@ -265,7 +265,8 @@ class Sender:
             if result and result.get("presigned_url"):
                 # Upload the bytes to the presigned URL
                 self._client.upload_media_bytes(
-                    result["presigned_url"], img_bytes, content_type
+                    result["presigned_url"], img_bytes, content_type,
+                    run_id=self._run_id, key=item["key"],
                 )
 
         elif media_type == "table":
@@ -293,7 +294,8 @@ class Sender:
             )
             if result and result.get("presigned_url"):
                 self._client.upload_media_bytes(
-                    result["presigned_url"], audio_bytes, content_type
+                    result["presigned_url"], audio_bytes, content_type,
+                    run_id=self._run_id, key=item["key"],
                 )
 
         elif media_type == "video":
@@ -311,7 +313,8 @@ class Sender:
             )
             if result and result.get("presigned_url"):
                 self._client.upload_media_bytes(
-                    result["presigned_url"], video_bytes, content_type
+                    result["presigned_url"], video_bytes, content_type,
+                    run_id=self._run_id, key=item["key"],
                 )
 
         elif media_type == "histogram":
@@ -408,7 +411,8 @@ class Sender:
             )
             if result and result.get("presigned_url"):
                 self._client.upload_media_bytes(
-                    result["presigned_url"], img_bytes, content_type
+                    result["presigned_url"], img_bytes, content_type,
+                    run_id=self._run_id, key=item["key"],
                 )
 
         elif media_type == "molecule":

{openrunner_sdk-2.2.0 → openrunner_sdk-2.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "openrunner-sdk"
-version = "2.2.0"
+version = "2.4.0"
 description = "OpenRunner SDK - W&B-compatible ML experiment tracking client"
 readme = "README.md"
 license = {text = "MIT"}

{openrunner_sdk-2.2.0 → openrunner_sdk-2.4.0}/tests/test_alert.py

@@ -192,7 +192,7 @@ class TestModuleLevelAlert:
             result = openrunner.alert("Title", text="Body", level="WARN")
             assert result == {"id": "x"}
             mock_run.alert.assert_called_once_with(
-                title="Title", text="Body", level="WARN"
+                title="Title", text="Body", level="WARN", wait_duration=None
             )
         finally:
             openrunner._active_run = original

{openrunner_sdk-2.2.0 → openrunner_sdk-2.4.0}/tests/test_integration_sklearn.py

@@ -70,15 +70,15 @@ class TestLogModel:
         from openrunner.integration.sklearn import log_model
         import openrunner
 
-        mock_log = MagicMock()
-        monkeypatch.setattr("openrunner.log", mock_log)
+        mock_config = MagicMock()
+        monkeypatch.setattr("openrunner.config", mock_config)
         monkeypatch.setattr("openrunner._active_run", MagicMock())
 
         model = _make_model(params={"n_estimators": 100, "max_depth": 5, "random_state": 42})
         log_model(model)
 
-        mock_log.assert_called_once()
-        logged = mock_log.call_args[0][0]
+        mock_config.update.assert_called_once()
+        logged = mock_config.update.call_args[0][0]
         assert logged["model/n_estimators"] == 100
         assert logged["model/max_depth"] == 5
         assert logged["model/random_state"] == 42
@@ -88,12 +88,12 @@ class TestLogModel:
         from openrunner.integration.sklearn import log_model
         import openrunner
 
-        mock_log = MagicMock()
-        monkeypatch.setattr("openrunner.log", mock_log)
+        mock_config = MagicMock()
+        monkeypatch.setattr("openrunner.config", mock_config)
         monkeypatch.setattr("openrunner._active_run", None)
 
         log_model(_make_model())
-        mock_log.assert_not_called()
+        mock_config.update.assert_not_called()
 
 
 class TestLogClassificationReport:

{openrunner_sdk-2.2.0 → openrunner_sdk-2.4.0}/tests/test_media.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import pytest
 import io
 import json
 import struct
@@ -586,16 +587,18 @@ class TestHtml:
     """Tests for the Html class."""
 
     def test_html_basic(self) -> None:
-        """Html with raw string serializes correctly."""
+        """Html with raw string serializes correctly (inject=True prepends style)."""
         html = Html("<div>Hello World</div>")
         result = html._serialize()
-        assert result == {"html": "<div>Hello World</div>"}
+        assert result["html"].endswith("<div>Hello World</div>")
+        assert "<style>" in result["html"]
 
     def test_html_with_caption(self) -> None:
         """Html with caption includes it in serialized output."""
         html = Html("<p>Report</p>", caption="Training Report")
         result = html._serialize()
-        assert result == {"html": "<p>Report</p>", "caption": "Training Report"}
+        assert result["html"].endswith("<p>Report</p>")
+        assert result["caption"] == "Training Report"
         assert html.caption == "Training Report"
 
     def test_html_no_caption(self) -> None:
@@ -616,7 +619,7 @@ class TestHtml:
         )
         html = Html(content, caption="complex")
         result = html._serialize()
-        assert result["html"] == content
+        assert result["html"].endswith(content)
         assert result["caption"] == "complex"
 
 
@@ -627,6 +630,11 @@ class TestHtml:
 class TestMatplotlibFigure:
     """Tests for the MatplotlibFigure class."""
 
+    pytestmark = pytest.mark.skipif(
+        not __import__("importlib").util.find_spec("matplotlib"),
+        reason="matplotlib not installed",
+    )
+
     def test_matplotlib_explicit_figure(self) -> None:
         """MatplotlibFigure from explicit figure serializes to PNG bytes."""
         import matplotlib

{openrunner_sdk-2.2.0 → openrunner_sdk-2.4.0}/tests/test_plot.py

@@ -208,6 +208,11 @@ class TestLineSeries:
 # pr_curve()
 # ---------------------------------------------------------------------------
 
+import importlib.util
+_has_sklearn = importlib.util.find_spec("sklearn") is not None
+
+
+@pytest.mark.skipif(not _has_sklearn, reason="scikit-learn not installed")
 class TestPRCurve:
     """Tests for plot.pr_curve()."""
 
@@ -272,6 +277,7 @@ class TestPRCurve:
 # roc_curve()
 # ---------------------------------------------------------------------------
 
+@pytest.mark.skipif(not _has_sklearn, reason="scikit-learn not installed")
 class TestROCCurve:
     """Tests for plot.roc_curve()."""
 
@@ -336,6 +342,7 @@ class TestROCCurve:
 # confusion_matrix()
 # ---------------------------------------------------------------------------
 
+@pytest.mark.skipif(not _has_sklearn, reason="scikit-learn not installed")
 class TestConfusionMatrix:
     """Tests for plot.confusion_matrix()."""
 

{openrunner_sdk-2.2.0 → openrunner_sdk-2.4.0}/tests/test_system_metrics.py

@@ -82,8 +82,8 @@ class TestGPUMetrics:
             assert result["system.gpu.0.gpu"] == 75.0
             assert "system.gpu.0.memory" in result
             assert result["system.gpu.0.memory"] == 50.0
-            assert "system.gpu.0.memoryAllocatedBytes" in result
-            assert result["system.gpu.0.memoryAllocatedBytes"] == 4_000_000_000.0
+            assert "system.gpu.0.memoryAllocatedMB" in result
+            assert result["system.gpu.0.memoryAllocatedMB"] == 4_000_000_000.0 / (1024 * 1024)
 
 
 # ---------------------------------------------------------------------------