hafnia 0.4.1__tar.gz → 0.4.2__tar.gz

{hafnia-0.4.1 → hafnia-0.4.2}/.github/workflows/Dockerfile

@@ -1,5 +1,8 @@
 ARG PYTHON_VERSION
 FROM python:${PYTHON_VERSION}-slim
+
+RUN apt-get update && apt-get install -y pigz && rm -rf /var/lib/apt/lists/*
+
 WORKDIR /opt/ml/processing
 
 COPY dist/*.whl .
@@ -8,4 +11,4 @@ RUN pip install --no-cache-dir *.whl && \
 
 RUN mkdir -p /opt/ml/processing/input \
              /opt/ml/processing/output \
-             /opt/ml/processing/tmp
+             /opt/ml/processing/tmp

{hafnia-0.4.1 → hafnia-0.4.2}/.github/workflows/build.yaml

@@ -6,6 +6,10 @@ on:
       python-version-file:
         required: true
         type: string
+      pyproject-toml-file:
+        required: false
+        type: string
+        default: "pyproject.toml"
     outputs:
       package-version:
         description: "The extracted package version"
@@ -25,7 +29,7 @@ jobs:
       - name: Extract package version
         id: extract-version
         run: |
-          VERSION=$(grep -m 1 'version = '  ${{ inputs.python-version-file }} | sed -e 's/version = "\(.*\)"/\1/')
+          VERSION=$(grep -m 1 'version = '  ${{ inputs.pyproject-toml-file }} | sed -e 's/version = "\(.*\)"/\1/')
           echo "package_version=$VERSION" >> $GITHUB_OUTPUT
 
       - name: Install uv

{hafnia-0.4.1 → hafnia-0.4.2}/.github/workflows/ci_cd.yaml

@@ -5,14 +5,14 @@ on:
     branches: [main]
   pull_request:
     branches: [main]
-    paths: ["src/**", "tests/**", "pyproject.toml"]
+    paths: ["src/**", "tests/**", "pyproject.toml", ".python-version", "uv.lock", ".github/workflows/**"]
 
 jobs:
   lint:
     name: Lint Code
     uses: ./.github/workflows/lint.yaml
     with:
-        python-version-file: "pyproject.toml"
+        python-version-file: ".python-version"
 
   security-scan:
     name: Security Scan
@@ -36,14 +36,15 @@ jobs:
     secrets: inherit
     uses: ./.github/workflows/tests.yaml
     with:
-        python-version-file: "pyproject.toml"
+        python-version-file: ".python-version"
 
   build:
     name: Build Package
     needs: [test, security-scan]
     uses: ./.github/workflows/build.yaml
     with:
-        python-version-file: "pyproject.toml"
+        python-version-file: ".python-version"
+        pyproject-toml-file: "pyproject.toml"
 
   publish-docker-staging:
     name: Publish Docker Image to Staging
@@ -52,7 +53,7 @@ jobs:
     if: github.event_name == 'push' && github.ref == 'refs/heads/main'
     uses: ./.github/workflows/publish_docker.yaml
     with:
-      python-version-file: "pyproject.toml"
+      python-version-file: ".python-version"
       package-version: ${{ needs.build.outputs.package-version }}
       environment: "staging"
 
@@ -80,7 +81,7 @@ jobs:
     if: github.event_name == 'push' && github.ref == 'refs/heads/main'
     uses: ./.github/workflows/publish_docker.yaml
     with:
-      python-version-file: "pyproject.toml"
+      python-version-file: ".python-version"
       package-version: ${{ needs.build.outputs.package-version }}
       environment: "production"
 

{hafnia-0.4.1 → hafnia-0.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hafnia
-Version: 0.4.1
+Version: 0.4.2
 Summary: Python SDK for communication with Hafnia platform.
 Author-email: Milestone Systems <hafniaplatform@milestone.dk>
 License-File: LICENSE

{hafnia-0.4.1 → hafnia-0.4.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hafnia"
-version = "0.4.1"
+version = "0.4.2"
 description = "Python SDK for communication with Hafnia platform."
 readme = "README.md"
 authors = [
@@ -44,14 +44,14 @@ dev = [
 test = ["pytest>=8.3.4", "pre-commit>=4.2.0", "ruff>=0.9.1"]
 
 [project.scripts]
-hafnia = 'cli.__main__:main'
+hafnia = 'hafnia_cli.__main__:main'
 
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"
 
 [tool.hatch.build.targets.wheel]
-packages = ["src/cli", "src/hafnia"]
+packages = ["src/hafnia_cli", "src/hafnia"]
 
 [tool.uv]
 default-groups = ["test"]

{hafnia-0.4.1 → hafnia-0.4.2}/src/hafnia/dataset/dataset_recipe/dataset_recipe.py

@@ -98,8 +98,8 @@ class DatasetRecipe(Serializable):
     @staticmethod
     def from_recipe_id(recipe_id: str) -> "DatasetRecipe":
         """Loads a dataset recipe by id from the hafnia platform."""
-        from cli.config import Config
         from hafnia.platform.dataset_recipe import get_dataset_recipe_by_id
+        from hafnia_cli.config import Config
 
         cfg = Config()
         endpoint_dataset = cfg.get_platform_endpoint("dataset_recipes")
@@ -114,8 +114,8 @@ class DatasetRecipe(Serializable):
     @staticmethod
     def from_recipe_name(name: str) -> "DatasetRecipe":
         """Loads a dataset recipe by name from the hafnia platform"""
-        from cli.config import Config
         from hafnia.platform.dataset_recipe import get_dataset_recipe_by_name
+        from hafnia_cli.config import Config
 
         cfg = Config()
         endpoint_dataset = cfg.get_platform_endpoint("dataset_recipes")
@@ -239,8 +239,8 @@ class DatasetRecipe(Serializable):
 
     def as_platform_recipe(self, recipe_name: Optional[str], overwrite: bool = False) -> Dict:
         """Uploads dataset recipe to the hafnia platform."""
-        from cli.config import Config
         from hafnia.platform.dataset_recipe import get_or_create_dataset_recipe
+        from hafnia_cli.config import Config
 
         recipe = self.as_dict()
         cfg = Config()

{hafnia-0.4.1 → hafnia-0.4.2}/src/hafnia/dataset/dataset_upload_helper.py

@@ -11,7 +11,6 @@ import polars as pl
 from PIL import Image
 from pydantic import BaseModel, ConfigDict, field_validator
 
-from cli.config import Config
 from hafnia.dataset import primitives
 from hafnia.dataset.dataset_names import (
     DatasetVariant,
@@ -33,6 +32,7 @@ from hafnia.dataset.primitives.primitive import Primitive
 from hafnia.http import post
 from hafnia.log import user_logger
 from hafnia.platform.datasets import get_dataset_id
+from hafnia_cli.config import Config
 
 
 def generate_bucket_name(dataset_name: str, deployment_stage: DeploymentStage) -> str:

{hafnia-0.4.1 → hafnia-0.4.2}/src/hafnia/platform/datasets.py

@@ -11,7 +11,6 @@ import rich
 from rich import print as rprint
 from rich.progress import track
 
-from cli.config import Config
 from hafnia import http, utils
 from hafnia.dataset.dataset_names import DATASET_FILENAMES_REQUIRED
 from hafnia.dataset.dataset_recipe.dataset_recipe import (
@@ -23,6 +22,7 @@ from hafnia.http import fetch
 from hafnia.log import sys_logger, user_logger
 from hafnia.platform.download import get_resource_credentials
 from hafnia.utils import timed
+from hafnia_cli.config import Config
 
 
 @timed("Fetching dataset list.")

{hafnia-0.4.1 → hafnia-0.4.2}/src/hafnia/utils.py

@@ -207,7 +207,7 @@ def is_hafnia_configured() -> bool:
     """
     Check if Hafnia is configured by verifying if the API key is set.
     """
-    from cli.config import Config
+    from hafnia_cli.config import Config
 
     return Config().is_configured()
 

{hafnia-0.4.1/src/cli → hafnia-0.4.2/src/hafnia_cli}/__main__.py

@@ -2,7 +2,7 @@
 import click
 
 import hafnia
-from cli import (
+from hafnia_cli import (
     consts,
     dataset_cmds,
     dataset_recipe_cmds,
@@ -11,7 +11,7 @@ from cli import (
     runc_cmds,
     trainer_package_cmds,
 )
-from cli.config import Config, ConfigSchema
+from hafnia_cli.config import Config, ConfigSchema
 
 
 @click.group()

{hafnia-0.4.1/src/cli → hafnia-0.4.2/src/hafnia_cli}/config.py

@@ -5,8 +5,8 @@ from typing import Dict, List, Optional
 
 from pydantic import BaseModel, field_validator
 
-import cli.consts as consts
-import cli.keychain as keychain
+import hafnia_cli.consts as consts
+import hafnia_cli.keychain as keychain
 from hafnia.log import sys_logger, user_logger
 
 PLATFORM_API_MAPPING = {

{hafnia-0.4.1/src/cli → hafnia-0.4.2/src/hafnia_cli}/dataset_cmds.py

@@ -3,9 +3,9 @@ from typing import Optional
 
 import click
 
-from cli import consts
-from cli.config import Config
 from hafnia import utils
+from hafnia_cli import consts
+from hafnia_cli.config import Config
 
 
 @click.group()

{hafnia-0.4.1/src/cli → hafnia-0.4.2/src/hafnia_cli}/dataset_recipe_cmds.py

@@ -4,7 +4,7 @@ from typing import Dict, Optional
 import click
 from rich import print as rprint
 
-from cli.config import Config
+from hafnia_cli.config import Config
 
 
 @click.group(name="dataset-recipe")

{hafnia-0.4.1/src/cli → hafnia-0.4.2/src/hafnia_cli}/experiment_cmds.py

@@ -3,7 +3,6 @@ from typing import Dict, Optional
 
 import click
 
-from cli.config import Config
 from hafnia import utils
 from hafnia.platform.dataset_recipe import (
     get_dataset_recipe_by_id,
@@ -11,6 +10,7 @@ from hafnia.platform.dataset_recipe import (
     get_or_create_dataset_recipe_by_dataset_name,
 )
 from hafnia.platform.trainer_package import create_trainer_package
+from hafnia_cli.config import Config
 
 
 @click.group(name="experiment")

{hafnia-0.4.1/src/cli → hafnia-0.4.2/src/hafnia_cli}/profile_cmds.py

@@ -2,8 +2,8 @@ import click
 from rich.console import Console
 from rich.table import Table
 
-import cli.consts as consts
-from cli.config import Config, ConfigSchema
+import hafnia_cli.consts as consts
+from hafnia_cli.config import Config, ConfigSchema
 
 
 @click.group()

{hafnia-0.4.1/src/cli → hafnia-0.4.2/src/hafnia_cli}/runc_cmds.py

@@ -7,8 +7,8 @@ from typing import Optional
 
 import click
 
-from cli.config import Config
 from hafnia.log import sys_logger, user_logger
+from hafnia_cli.config import Config
 
 
 @click.group(name="runc")

{hafnia-0.4.1/src/cli → hafnia-0.4.2/src/hafnia_cli}/trainer_package_cmds.py

@@ -3,8 +3,8 @@ from typing import Optional
 
 import click
 
-import cli.consts as consts
-from cli.config import Config
+import hafnia_cli.consts as consts
+from hafnia_cli.config import Config
 
 
 @click.group(name="trainer")

{hafnia-0.4.1 → hafnia-0.4.2}/tests/integration/test_cli_integration.py

@@ -13,7 +13,7 @@ def test_cli_integration_test():
     Run actual CLI commands against Hafnia platform to ensure end-to-end functionality
     using simple smoke tests
     """
-    from cli.__main__ import main as hafnia_cli
+    from hafnia_cli.__main__ import main as hafnia_cli
 
     if not is_hafnia_configured():
         pytest.skip("Hafnia platform not configured. Skipping CLI integration test.")

{hafnia-0.4.1 → hafnia-0.4.2}/tests/integration/test_dataset_recipes_with_platform.py

@@ -1,7 +1,6 @@
 import pytest
 import urllib3
 
-from cli.config import Config
 from hafnia.dataset.dataset_names import OPS_REMOVE_CLASS
 from hafnia.dataset.dataset_recipe.dataset_recipe import (
     DatasetRecipe,
@@ -12,6 +11,7 @@ from hafnia.platform.dataset_recipe import (
     get_dataset_recipe_by_id,
 )
 from hafnia.utils import is_hafnia_configured
+from hafnia_cli.config import Config
 
 
 def test_dataset_recipe_on_platform():

{hafnia-0.4.1 → hafnia-0.4.2}/tests/unit/test_cli.py

@@ -4,9 +4,9 @@ from typing import Dict
 import pytest
 from click.testing import CliRunner
 
-import cli.__main__ as cli
-import cli.consts as consts
-from cli.config import Config, ConfigFileSchema, ConfigSchema
+import hafnia_cli.__main__ as cli
+import hafnia_cli.consts as consts
+from hafnia_cli.config import Config, ConfigFileSchema, ConfigSchema
 
 
 @pytest.fixture
@@ -99,13 +99,13 @@ class TestProfile:
     ) -> None:
         """Test list of profiles functionality."""
         with pytest.MonkeyPatch.context() as mp:
-            mp.setattr("cli.__main__.Config", lambda *args, **kwargs: empty_config)
+            mp.setattr("hafnia_cli.__main__.Config", lambda *args, **kwargs: empty_config)
             result = cli_runner.invoke(cli.main, ["profile", "ls"])
             assert result.exit_code != 0
             assert consts.ERROR_CONFIGURE in result.output
 
         with pytest.MonkeyPatch.context() as mp:
-            mp.setattr("cli.__main__.Config", lambda *args, **kwargs: config_with_profiles)
+            mp.setattr("hafnia_cli.__main__.Config", lambda *args, **kwargs: config_with_profiles)
             result = cli_runner.invoke(cli.main, ["profile", "ls"])
             assert result.exit_code == 0
             assert "default" in result.output
@@ -115,13 +115,13 @@ class TestProfile:
 
     def test_switch_profile(self, cli_runner: CliRunner, empty_config: Config, config_with_profiles: Config) -> None:
         with pytest.MonkeyPatch.context() as mp:
-            mp.setattr("cli.__main__.Config", lambda *args, **kwargs: empty_config)
+            mp.setattr("hafnia_cli.__main__.Config", lambda *args, **kwargs: empty_config)
             result = cli_runner.invoke(cli.main, ["profile", "use", "default"])
             assert result.exit_code != 0
             assert f"Error: {consts.ERROR_CONFIGURE}" in result.output
 
         with pytest.MonkeyPatch.context() as mp:
-            mp.setattr("cli.__main__.Config", lambda *args, **kwargs: config_with_profiles)
+            mp.setattr("hafnia_cli.__main__.Config", lambda *args, **kwargs: config_with_profiles)
             result = cli_runner.invoke(cli.main, ["profile", "active"])
             assert result.exit_code == 0
             assert f"{consts.PROFILE_TABLE_HEADER} default" in result.output
@@ -140,13 +140,13 @@ class TestProfile:
 
     def test_remove_profile(self, cli_runner: CliRunner, empty_config: Config, config_with_profiles: Config) -> None:
         with pytest.MonkeyPatch.context() as mp:
-            mp.setattr("cli.__main__.Config", lambda *args, **kwargs: empty_config)
+            mp.setattr("hafnia_cli.__main__.Config", lambda *args, **kwargs: empty_config)
             result = cli_runner.invoke(cli.main, ["profile", "rm", "default"])
             assert result.exit_code != 0
             assert consts.ERROR_CONFIGURE in result.output
 
         with pytest.MonkeyPatch.context() as mp:
-            mp.setattr("cli.__main__.Config", lambda *args, **kwargs: config_with_profiles)
+            mp.setattr("hafnia_cli.__main__.Config", lambda *args, **kwargs: config_with_profiles)
             result = cli_runner.invoke(cli.main, ["profile", "rm", "staging"])
             assert result.exit_code == 0
             assert f"{consts.PROFILE_REMOVED_SUCCESS} staging" in result.output

{hafnia-0.4.1 → hafnia-0.4.2}/uv.lock

@@ -1076,7 +1076,7 @@ wheels = [
 
 [[package]]
 name = "hafnia"
-version = "0.4.1"
+version = "0.4.2"
 source = { editable = "." }
 dependencies = [
     { name = "boto3" },