modelzone-sdk 0.1.0__tar.gz → 0.2.0.dev0__tar.gz

{modelzone_sdk-0.1.0 → modelzone_sdk-0.2.0.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: modelzone-sdk
-Version: 0.1.0
+Version: 0.2.0.dev0
 Summary: Modelzone SDK – a slim model training and serving toolkit
 License-Expression: Apache-2.0
 Author: Team Enigma
@@ -14,10 +14,8 @@ Provides-Extra: azureml
 Provides-Extra: training
 Requires-Dist: azure-ai-ml ; extra == "azureml"
 Requires-Dist: azure-identity ; extra == "azureml"
-Requires-Dist: datamazing (>=7,<8)
 Requires-Dist: deltalake (>=1) ; extra == "training"
 Requires-Dist: mlflow (>=3) ; extra == "azureml"
-Requires-Dist: pandas (>=2,<3)
 Description-Content-Type: text/markdown
 
 # Modelzone SDK
@@ -114,3 +112,4 @@ modelzone fetch my_model_package --output-dir .model
 poetry install --with dev,test
 pytest
 ```
+

{modelzone_sdk-0.1.0 → modelzone_sdk-0.2.0.dev0}/README.md

@@ -91,4 +91,4 @@ modelzone fetch my_model_package --output-dir .model
 ```bash
 poetry install --with dev,test
 pytest
-```
+```

{modelzone_sdk-0.1.0 → modelzone_sdk-0.2.0.dev0}/modelzone/azureml/backend.py

@@ -18,15 +18,11 @@ import tempfile
 from typing import Any
 
 import datamazing.pandas as pdz
-import requests as _requests
+import requests
 
-from modelzone.core.constants import CODE_DIR, RECORD_FILENAME, TRAINED_MODEL_FILENAME
-from modelzone.core.model import Model
-from modelzone.core.record import RunRecord
-from modelzone.core.trained_model import TrainedModel
-from modelzone.core.training_result import TrainingResult
-from modelzone.training.backend import Backend
-from modelzone.training.context import TrainingContext
+from modelzone.core.constants import RECORD_FILENAME, TRAINED_MODEL_FILENAME
+from modelzone.core import Model, RunRecord, TrainedModel, TrainingResult
+from modelzone.training import Backend, TrainingContext
 
 try:
     import mlflow
@@ -174,7 +170,7 @@ class AzureMLBackend(Backend):
             f"/runs/{run_id}/artifacts/batch/metadata"
         )
         body = {"paths": [{"path": p} for p in remote_paths]}
-        resp = _requests.post(
+        resp = requests.post(
             url,
             json=body,
             headers={
@@ -199,7 +195,7 @@ class AzureMLBackend(Backend):
             f"{self._history_base()}/experiments/{self.experiment_name}"
             f"/runs/{run_id}/artifacts"
         )
-        resp = _requests.get(url, headers=self._auth_headers())
+        resp = requests.get(url, headers=self._auth_headers())
         resp.raise_for_status()
         all_artifacts = resp.json().get("value", [])
 
@@ -216,7 +212,7 @@ class AzureMLBackend(Backend):
                 f"{self._history_base()}/experiments/{self.experiment_name}"
                 f"/runs/{run_id}/artifacts/contentinfo"
             )
-            resp = _requests.get(
+            resp = requests.get(
                 info_url,
                 params={"path": remote_path},
                 headers=self._auth_headers(),
@@ -439,8 +435,7 @@ class AzureMLBackend(Backend):
             download_path=download_dir,
         )
 
-        # Downloaded files are nested under <model_name>/model/
-        local_path = os.path.join(download_dir, model_name, "model")
+        local_path = os.path.join(download_dir, model_name, "run_output")
 
         with open(os.path.join(local_path, TRAINED_MODEL_FILENAME), "rb") as f:
             trained_model: TrainedModel = pickle.load(f)  # noqa: S301

{modelzone_sdk-0.1.0 → modelzone_sdk-0.2.0.dev0}/modelzone/training/backend.py

@@ -14,10 +14,7 @@ from typing import Any
 import datamazing.pandas as pdz
 
 from modelzone.core.constants import CODE_DIR, RECORD_FILENAME, TRAINED_MODEL_FILENAME
-from modelzone.core.model import Model
-from modelzone.core.record import RunRecord
-from modelzone.core.trained_model import TrainedModel
-from modelzone.core.training_result import TrainingResult
+from modelzone.core import Model, RunRecord, TrainedModel, TrainingResult
 from modelzone.training.context import TrainingContext
 
 _SNAPSHOT_IGNORE = shutil.ignore_patterns(
@@ -32,6 +29,7 @@ _SNAPSHOT_IGNORE = shutil.ignore_patterns(
     "node_modules",
     "build",
     "dist",
+    ".cache",
 )
 
 

{modelzone_sdk-0.1.0 → modelzone_sdk-0.2.0.dev0}/modelzone/training/context.py

@@ -11,12 +11,17 @@ from __future__ import annotations
 
 import json
 import os
-from typing import Any
-
-import datamazing.pandas as pdz
+from typing import Any, Protocol
 
 from modelzone.core.record import Metric, RunRecord
 
+
+class Database(Protocol):
+    """Structural type for a database that can execute queries."""
+
+    def query(self, table_name: str, **kwargs: Any) -> Any: ...
+
+
 # Subdirectories / files inside the run directory
 _METRICS_FILE = "metrics.jsonl"
 _LOG_FILE = "output.log"
@@ -43,7 +48,7 @@ class TrainingContext:
         run_dir: str,
         seed: int,
         params: dict[str, Any] | None = None,
-        db: pdz.Database | None = None,
+        db: Database | None = None,
     ) -> None:
         self.run_dir = run_dir
         self.seed = seed

{modelzone_sdk-0.1.0 → modelzone_sdk-0.2.0.dev0}/pyproject.toml

@@ -1,12 +1,12 @@
 [project]
 name = "modelzone-sdk"
-version = "0.1.0"
+version = "0.2.0.dev"
 description = "Modelzone SDK – a slim model training and serving toolkit"
 authors = [{ name = "Team Enigma", email = "enigma@energinet.dk" }]
 requires-python = ">=3.13"
 readme = "README.md"
 license = "Apache-2.0"
-dependencies = ["datamazing>=7,<8", "pandas>=2,<3"]
+dependencies = []
 classifiers = [
     "Development Status :: 3 - Alpha",
     "Intended Audience :: Developers",
@@ -26,8 +26,6 @@ packages = [{ include = "modelzone" }]
 requires-poetry = ">=2.2"
 
 [tool.poetry.dependencies]
-datamazing = "^7.0.1"
-pandas = ">=2.3.3,<3"
 
 [tool.poetry.group.training.dependencies]
 deltalake = ">=1"