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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: modelzone-sdk
-Version: 0.1.1.dev0
+Version: 0.2.0
 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.1.dev0 → modelzone_sdk-0.2.0}/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.1.dev0 → modelzone_sdk-0.2.0}/modelzone/azureml/backend.py

@@ -17,9 +17,10 @@ import shutil
 import tempfile
 from typing import Any
 
-import datamazing.pandas as pdz
 import requests
 
+from modelzone.core.protocols import Database
+
 from modelzone.core.constants import RECORD_FILENAME, TRAINED_MODEL_FILENAME
 from modelzone.core import Model, RunRecord, TrainedModel, TrainingResult
 from modelzone.training import Backend, TrainingContext
@@ -245,7 +246,7 @@ class AzureMLBackend(Backend):
         seed: int | None = None,
         params: dict[str, Any] | None = None,
         tags: dict[str, str] | None = None,
-        db: pdz.Database | None = None,
+        db: Database | None = None,
         source_dir: str | None = None,
     ) -> TrainingResult:
         """Execute a training run tracked by AzureML via MLflow."""
@@ -293,7 +294,7 @@ class AzureMLBackend(Backend):
         run_dir: str,
         seed: int,
         params: dict[str, Any] | None,
-        db: pdz.Database | None,
+        db: Database | None,
     ) -> _AzureMLTrainingContext:
         """Return an MLflow-aware context and log params/seed eagerly."""
         if params:

{modelzone_sdk-0.1.1.dev0 → modelzone_sdk-0.2.0}/modelzone/core/predict_context.py

@@ -3,8 +3,8 @@
 from __future__ import annotations
 
 from typing import Any
-import datamazing.pandas as pdz
 
+from modelzone.core.protocols import Database, TimeInterval
 from modelzone.core.trained_model import TrainedModel
 
 
@@ -23,8 +23,8 @@ class PredictContext:
     def __init__(
         self,
         trained_model: TrainedModel,
-        db: pdz.Database,
-        time_interval: pdz.TimeInterval | None = None,
+        db: Database,
+        time_interval: TimeInterval | None = None,
         identifier: str | None = None,
     ) -> None:
         self.trained_model = trained_model

modelzone_sdk-0.2.0/modelzone/core/protocols.py

@@ -0,0 +1,25 @@
+"""Structural (Protocol) types shared across modelzone.
+
+These exist so that modules can annotate parameters that accept
+datamazing.pandas.Database / TimeInterval (or any structurally
+compatible object) without depending on the datamazing package itself.
+"""
+
+from __future__ import annotations
+
+from typing import Any, Protocol, runtime_checkable
+
+
+@runtime_checkable
+class TimeInterval(Protocol):
+    """Any object exposing ``.left`` and ``.right`` bound attributes."""
+
+    left: Any
+    right: Any
+
+
+@runtime_checkable
+class Database(Protocol):
+    """Structural type for a database that can execute queries."""
+
+    def query(self, table_name: str, **kwargs: Any) -> Any: ...

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

@@ -11,8 +11,7 @@ import uuid
 from abc import ABC, abstractmethod
 from typing import Any
 
-import datamazing.pandas as pdz
-
+from modelzone.core.protocols import Database
 from modelzone.core.constants import CODE_DIR, RECORD_FILENAME, TRAINED_MODEL_FILENAME
 from modelzone.core import Model, RunRecord, TrainedModel, TrainingResult
 from modelzone.training.context import TrainingContext
@@ -55,7 +54,7 @@ class Backend(ABC):
         seed: int | None = None,
         params: dict[str, Any] | None = None,
         tags: dict[str, str] | None = None,
-        db: pdz.Database | None = None,
+        db: Database | None = None,
         source_dir: str | None = None,
     ) -> TrainingResult:
         """Execute a training run.
@@ -126,7 +125,7 @@ class Backend(ABC):
         run_dir: str,
         seed: int,
         params: dict[str, Any] | None,
-        db: pdz.Database | None,
+        db: Database | None,
     ) -> TrainingContext:
         """Create the :class:`TrainingContext` for a training run.
 

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

@@ -13,10 +13,10 @@ import json
 import os
 from typing import Any
 
-import datamazing.pandas as pdz
-
+from modelzone.core.protocols import Database
 from modelzone.core.record import Metric, RunRecord
 
+
 # Subdirectories / files inside the run directory
 _METRICS_FILE = "metrics.jsonl"
 _LOG_FILE = "output.log"
@@ -43,7 +43,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.1.dev0 → modelzone_sdk-0.2.0}/modelzone/training/db.py

@@ -1,8 +1,8 @@
 """Modelzone training – database client for accessing training data.
 
 Wraps a :class:`~modelzone.training.delta_client.DeltaClient` and
-implements :class:`datamazing.pandas.Database` so it can be used
-interchangeably wherever a ``pdz.Database`` is expected (including
+implements :class:`~modelzone.core.protocols.Database` so it can be used
+interchangeably wherever a ``Database`` is expected (including
 :class:`~modelzone.core.PredictContext`).
 """
 
@@ -13,17 +13,17 @@ import json
 import os
 from typing import Optional
 
-import datamazing.pandas as pdz
 import pandas as pd
 
+from modelzone.core.protocols import TimeInterval
 from modelzone.training.delta_client import DeltaClient
 
 
-class ModelzoneDatabase(pdz.Database):
+class ModelzoneDatabase:
     """Database client for loading training (and prediction) data.
 
     Thin wrapper around :class:`DeltaClient` that
-    implements the :class:`datamazing.pandas.Database` ABC.
+    implements the :class:`~modelzone.core.protocols.Database` protocol.
 
     When *cache_dir* is set, query results are saved as local parquet
     files keyed by the query parameters.  This avoids repeated reads
@@ -88,7 +88,7 @@ class ModelzoneDatabase(pdz.Database):
     @staticmethod
     def _cache_key(
         table_name: str,
-        time_interval: pdz.TimeInterval | None,
+        time_interval: TimeInterval | None,
         filters: dict[str, object] | None,
         columns: list[str] | None,
     ) -> str:
@@ -108,19 +108,19 @@ class ModelzoneDatabase(pdz.Database):
         return hashlib.sha256(blob).hexdigest()[:16]
 
     # ------------------------------------------------------------------
-    # pdz.Database interface
+    # Database protocol interface
     # ------------------------------------------------------------------
 
     def query(
         self,
         table_name: str,
-        time_interval: Optional[pdz.TimeInterval] = None,
+        time_interval: Optional[TimeInterval] = None,
         filters: Optional[dict[str, object]] = None,
         columns: Optional[list[str]] = None,
     ) -> pd.DataFrame:
         """Query a table and return a DataFrame.
 
-        Implements :meth:`datamazing.pandas.Database.query`.
+        Implements :meth:`~modelzone.core.protocols.Database.query`.
         When *cache_dir* is configured, results are read from / written
         to a local parquet file to speed up repeated queries.
 

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

@@ -1,12 +1,12 @@
 [project]
 name = "modelzone-sdk"
-version = "0.1.1.dev"
+version = "0.2.0"
 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",
@@ -25,17 +25,6 @@ modelzone = "modelzone.cli:main"
 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"
-
-[tool.poetry.group.dev.dependencies]
-pre-commit = ">=2.20.0"
-ruff = ">=0.4"
-
 [tool.poetry.group.test.dependencies]
 pytest = ">=8"
 pytest-cov = ">=3.0.0"
@@ -48,13 +37,6 @@ build-backend = "poetry.core.masonry.api"
 [tool.black]
 line-length = 88
 
-[tool.ruff]
-line-length = 88
-target-version = "py313"
-
-[tool.ruff.lint]
-select = ["E", "F", "I", "W"]
-
 [tool.mypy]
 ignore_missing_imports = true