hafnia 0.1.8__tar.gz → 0.1.9__tar.gz

{hafnia-0.1.8 → hafnia-0.1.9}/.devcontainer/devcontainer.json

@@ -1,5 +1,5 @@
 {
-  "name": "MDI Tools Container",
+  "name": "Hafnia Tools Container",
   "image": "mcr.microsoft.com/devcontainers/python:1-3.10-bullseye",
   "features": {
     "ghcr.io/jsburckhardt/devcontainer-features/uv:1": {},

{hafnia-0.1.8 → hafnia-0.1.9}/.github/workflows/ci_cd.yaml

@@ -45,7 +45,7 @@ jobs:
         python-version-file: "pyproject.toml"
 
   publish-pypi:
-    name: Publish Package
+    name: Publish Package to PyPI
     needs: build
     if: github.event_name == 'push' && github.ref == 'refs/heads/main'
     uses: ./.github/workflows/publish_pypi.yaml
@@ -53,7 +53,7 @@ jobs:
       environment: pypi
 
   publish-pypi-test:
-    name: Publish Package
+    name: Publish Package to TestPyPI
     needs: build
     if: github.event_name == 'push' && github.ref == 'refs/heads/main'
     uses: ./.github/workflows/publish_pypi.yaml

{hafnia-0.1.8 → hafnia-0.1.9}/.vscode/launch.json

@@ -22,7 +22,7 @@
             ],
         },
         {
-            "name": "debug (mdi data download mnist)",
+            "name": "debug (hafnia data download mnist)",
             "type": "debugpy",
             "request": "launch",
             "program": "${workspaceFolder}/src/cli/__main__.py",

{hafnia-0.1.8 → hafnia-0.1.9}/.vscode/settings.json

@@ -4,7 +4,7 @@
         "editor.defaultFormatter": "vscode.json-language-features"
     },
     "editor.rulers": [
-        100,  // This should match the 'line-length' setting in pyproject.toml
+        120,  // This should match the 'line-length' setting in pyproject.toml
     ],
     "notebook.formatOnSave.enabled": true,
     "notebook.codeActionsOnSave": {

{hafnia-0.1.8 → hafnia-0.1.9}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: hafnia
-Version: 0.1.8
-Summary: Python tools for communication with MDI platform.
+Version: 0.1.9
+Summary: Python tools for communication with Hafnia platform.
 Author-email: Ivan Sahumbaiev <ivsa@milestone.dk>
 Requires-Python: >=3.10
 Requires-Dist: boto3>=1.35.91
@@ -25,7 +25,7 @@ Project Hafnia is a comprehensive solution for managing data science experiments
 
 ## Documentation
 
-- [CLI Documentation](docs/cli.md) - Detailed guide for the MDI command-line interface
+- [CLI Documentation](docs/cli.md) - Detailed guide for the Hafnia command-line interface
 - [Script2Model Documentation](docs/s2m.md) - Detailed guide for script2model
 - [Release lyfecycle](docs/release.md) - Details about package release lifecycle.
 
@@ -38,6 +38,5 @@ Project Hafnia is a comprehensive solution for managing data science experiments
 ## Install
 
 ```bash
-pip install -i https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple/hafnia
+pip install hafnia
 ```
-

{hafnia-0.1.8 → hafnia-0.1.9}/README.md

@@ -5,7 +5,7 @@ Project Hafnia is a comprehensive solution for managing data science experiments
 
 ## Documentation
 
-- [CLI Documentation](docs/cli.md) - Detailed guide for the MDI command-line interface
+- [CLI Documentation](docs/cli.md) - Detailed guide for the Hafnia command-line interface
 - [Script2Model Documentation](docs/s2m.md) - Detailed guide for script2model
 - [Release lyfecycle](docs/release.md) - Details about package release lifecycle.
 
@@ -18,6 +18,5 @@ Project Hafnia is a comprehensive solution for managing data science experiments
 ## Install
 
 ```bash
-pip install -i https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple/hafnia
+pip install hafnia
 ```
-

hafnia-0.1.9/docs/cli.md

@@ -0,0 +1,98 @@
+A command-line interface tool for managing data science experiments and resources on the Project Hafnia.
+
+## Features
+
+- **Platform Configuration**: Easy setup and management of Hafnia platform settings
+
+## Installation
+
+## CLI Commands
+
+### Core Commands
+
+- `hafnia configure` - Configure Hafnia CLI settings
+- `hafnia clear` - Remove stored configuration
+- `hafnia profile` - Manage profiles (see subcommands below) 
+
+### Profile Management
+
+- `hafnia profile ls` - List all available profiles
+- `hafnia profile use <profile_name>` - Switch to a different profile
+- `hafnia profile rm <profile_name>` - Remove a specific profile
+- `hafnia profile active` - Show detailed information about the active profile
+
+### Data Management
+
+- `hafnia data get <url> <destination>` - Download resource from Hafnia platform to local destination
+
+### Experiment Management
+
+- `hafnia runc launch <task>` - Launch a job within the image
+- `hafnia runc build <recipe_url> [state_file] [ecr_repository] [image_name]` - Build docker image with a given recipe
+- `hafnia runc build-local <recipe> [state_file] [image_name]` - Build recipe from local path as image with prefix - localhost
+
+## Configuration
+
+he CLI tool supports multiple configuration profiles:
+
+1. Run `hafnia configure`
+2. Enter a profile name (defaults to "default")
+3. Enter your Hafnia API Key when prompted
+4. Provide the Hafnia Platform URL (defaults to "https://api.mdi.milestonesys.com")
+5. The organization ID will be retrieved automatically
+6. Verify your configuration with `hafnia profile active`
+
+## Example Usage
+
+```bash
+# Configure the CLI with a new profile
+hafnia configure
+
+# List all available profiles
+hafnia profile ls
+
+# Switch to a different profile
+hafnia profile use production
+
+# View active profile details
+hafnia profile active
+
+# Remove a profile
+hafnia profile rm old-profile
+
+# Clear all configuration
+hafnia clear
+
+# Download a dataset sample
+hafnia data download mnist
+
+# Add '--force' to re-download dataset
+hafnia data download mnist --force
+
+# Download a resource from the platform
+hafnia data get https://api.mdi.milestonesys.com/api/v1/datasets/my-dataset ./data
+
+# Build a Docker image from a recipe
+hafnia runc build https://api.mdi.milestonesys.com/api/v1/recipes/my-recipe
+
+# Build a Docker image from a local recipe
+hafnia runc build-local ./my-recipe
+
+# Launch a task within the image
+hafnia runc launch train
+```
+
+## Environment Variables
+
+The CLI tool uses configuration stored in your local environment. You can view the current settings using:
+
+```bash
+hafnia profile active
+```
+
+Available environment variables:
+
+- `MDI_CONFIG_PATH` - Custom path to the configuration file
+- `MDI_API_KEY_SECRET_NAME` - Name of the AWS Secrets Manager secret containing the API key
+- `AWS_REGION` - AWS region for ECR and Secrets Manager operations
+- `RECIPE_DIR` - Directory containing recipe code (used by the `runc launch` command)

{hafnia-0.1.8 → hafnia-0.1.9}/docs/s2m.md

@@ -1,6 +1,6 @@
 # Script2Model: Converting Scripts to Models
 
-Script2Model is a utility in the MDI Python Tools that helps you convert your Python scripts into deployable models on the MDI platform.
+Script2Model is a utility in the Hafnia Python Tools that helps you convert your Python scripts into deployable models on the Hafnia platform.
 
 ## Overview
 
@@ -9,11 +9,11 @@ Script2Model simplifies the process of converting experimental code into product
 1. Analyzing your Python script
 2. Extracting the necessary dependencies and functionality
 3. Creating a standardized model structure
-4. Packaging the model for deployment on the MDI platform
+4. Packaging the model for deployment on the Hafnia platform
 
 ## Prerequisites
 
-- An active MDI platform profile (configure with `mdi configure`)
+- An active Hafnia platform profile (configure with `hafnia configure`)
 - Python scripts with well-defined inputs and outputs
 - Required dependencies installed
 
@@ -22,7 +22,7 @@ Script2Model simplifies the process of converting experimental code into product
 To convert a Python script to a model:
 
 ```bash
-mdi script2model convert <script_path> [--output <output_dir>] [--name <model_name>]
+hafnia script2model convert <script_path> [--output <output_dir>] [--name <model_name>]
 ```
 
 ### Parameters
@@ -35,10 +35,10 @@ mdi script2model convert <script_path> [--output <output_dir>] [--name <model_na
 
 ```bash
 # Convert a training script to a model
-mdi script2model convert ./src/scripts/train.py --name my-classifier
+hafnia script2model convert ./src/scripts/train.py --name my-classifier
 
-# Deploy the model to MDI platform
-mdi script2model deploy ./model --experiment-id exp-123456
+# Deploy the model to Hafnia platform
+hafnia script2model deploy ./model --experiment-id exp-123456
 ```
 
 ## Working with Model Configuration
@@ -47,10 +47,10 @@ Script2Model automatically detects function signatures and generates a configura
 
 ```bash
 # Generate a configuration template
-mdi script2model config-template ./src/scripts/train.py
+hafnia script2model config-template ./src/scripts/train.py
 
 # Convert with a custom configuration
-mdi script2model convert ./src/scripts/train.py --config ./config.json
+hafnia script2model convert ./src/scripts/train.py --config ./config.json
 ```
 
 ## Best Practices
@@ -61,15 +61,15 @@ mdi script2model convert ./src/scripts/train.py --config ./config.json
 4. **Dependencies**: List all dependencies in requirements.txt or environment.yml
 5. **Testing**: Test your script locally before conversion
 
-## Logging with MDILogger
+## Logging with HafniaLogger
 
-Script2Model integrates with MDILogger to track model training and evaluation metrics:
+Script2Model integrates with HafniaLogger to track model training and evaluation metrics:
 
 ```python
-from hafnia.experiment import MDILogger
+from hafnia.experiment import HafniaLogger
 
 # Initialize logger
-logger = MDILogger(Path("./logs"), update_interval=5)
+logger = HafniaLogger(Path("./logs"), update_interval=5)
 
 # Log metrics during training
 logger.log_metric("accuracy", value=0.95, step=100)

hafnia-0.1.9/examples/script2model/pytorch/Dockerfile

@@ -0,0 +1,10 @@
+FROM pytorch/pytorch:latest
+
+RUN pip install torchmetrics tqdm tensorboard
+
+## Hafnia section of the Dockerfile
+RUN pip install hafnia
+
+# Copy recipe files in the docker image
+ENV RECIPE_DIR=/opt/recipe
+COPY src $RECIPE_DIR

{hafnia-0.1.8 → hafnia-0.1.9}/examples/script2model/pytorch/src/lib/train_utils.py

@@ -8,7 +8,7 @@ from torchvision.models import resnet18
 from torchvision.transforms import v2
 
 from hafnia.data import load_dataset
-from hafnia.experiment import MDILogger
+from hafnia.experiment import HafniaLogger
 
 
 def create_transforms() -> v2.Compose:
@@ -43,10 +43,10 @@ def create_dataloaders(
         Tuple[DataLoader, DataLoader]: Training and testing DataLoaders.
     """
     transforms = create_transforms()
-    mdi_dataset = load_dataset(data_root)
-    train_split = mdi_dataset["train"]
+    hafnia_dataset = load_dataset(data_root)
+    train_split = hafnia_dataset["train"]
     train_split.set_transform(transforms)
-    test_split = mdi_dataset["test"]
+    test_split = hafnia_dataset["test"]
     test_split.set_transform(transforms)
 
     train_loader = DataLoader(
@@ -83,7 +83,7 @@ def run_train_epoch(
     criterion: nn.Module,
     metrics: nn.Module,
     device: torch.device,
-    ml_logger: MDILogger,
+    ml_logger: HafniaLogger,
     log_interval: int,
     max_steps_per_epoch: int,
 ) -> Dict[str, float]:
@@ -98,7 +98,7 @@ def run_train_epoch(
         criterion (nn.Module): Loss function.
         metrics (MulticlassAccuracy): Metrics calculator.
         device (torch.device): Computation device.
-        ml_logger (MDILogger): Logger for metrics.
+        ml_logger (HafniaLogger): Logger for metrics.
         log_interval (int): Interval for logging.
         max_steps_per_epoch (int): Maximum steps per epoch.
 
@@ -154,7 +154,7 @@ def run_eval(
     criterion: nn.Module,
     metrics: nn.Module,
     device: torch.device,
-    ml_logger: MDILogger,
+    ml_logger: HafniaLogger,
 ):
     """
     Runs evaluation on the test dataset.
@@ -166,7 +166,7 @@ def run_eval(
         criterion (nn.Module): Loss function.
         metrics (MulticlassAccuracy): Metrics calculator.
         device (torch.device): Computation device.
-        ml_logger (MDILogger): Logger for metrics.
+        ml_logger (HafniaLogger): Logger for metrics.
 
     Returns:
         Dict[str, float]: Dictionary containing average loss and accuracy.
@@ -201,7 +201,7 @@ def run_eval(
 
 
 def train_loop(
-    logger: MDILogger,
+    logger: HafniaLogger,
     train_dataloader: DataLoader,
     test_dataloader: DataLoader,
     model: nn.Module,
@@ -215,7 +215,7 @@ def train_loop(
     Main training loop.
 
     Args:
-        logger (MDILogger): Logger for metrics.
+        logger (HafniaLogger): Logger for metrics.
         train_dataloader (DataLoader): Training DataLoader.
         test_dataloader (DataLoader): Testing DataLoader.
         model (nn.Module): The model to train.

{hafnia-0.1.8 → hafnia-0.1.9}/examples/script2model/pytorch/src/scripts/train.py

@@ -7,7 +7,7 @@ from pathlib import Path
 import torch
 from train_utils import create_dataloaders, create_model, train_loop
 
-from hafnia.experiment import MDILogger
+from hafnia.experiment import HafniaLogger
 
 DATA_DIR = os.getenv("MDI_DATASET_DIR", "/opt/ml/input/data/training")
 ARTIFACT_DIR = os.getenv("MDI_ARTIFACT_DIR", "/opt/ml/output/data")
@@ -37,7 +37,7 @@ def main(args: argparse.Namespace):
     model_dir = Path(MODEL_DIR)
     model_dir.mkdir(parents=True, exist_ok=True)
 
-    logger = MDILogger(artifacts_dir)
+    logger = HafniaLogger(artifacts_dir)
     logger.log_configuration(vars(args))
     train_dataloader, test_dataloader = create_dataloaders(DATA_DIR, args.batch_size)
     model = create_model(num_classes=10)

{hafnia-0.1.8 → hafnia-0.1.9}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "hafnia"
-version = "0.1.8"
-description = "Python tools for communication with MDI platform."
+version = "0.1.9"
+description = "Python tools for communication with Hafnia platform."
 readme = "README.md"
 authors = [{ name = "Ivan Sahumbaiev", email = "ivsa@milestone.dk" }]
 requires-python = ">=3.10"
@@ -31,7 +31,7 @@ lint = [
 default-groups = "all"
 
 [project.scripts]
-mdi = 'cli.__main__:main'
+hafnia = 'cli.__main__:main'
 
 [project.optional-dependencies]
 # Use "uv sync --extra torch" to install torch dependencies
@@ -48,7 +48,7 @@ build-backend = "hatchling.build"
 [tool.ruff]
 select = ["I", "E", "F"]
 ignore = ["E501"]
-line-length = 100
+line-length = 120
 
 [tool.ruff.isort]
 relative-imports-order = "closest-to-furthest"

{hafnia-0.1.8 → hafnia-0.1.9}/src/cli/__main__.py

@@ -8,14 +8,14 @@ from cli.config import Config, ConfigSchema
 @click.group()
 @click.pass_context
 def main(ctx: click.Context) -> None:
-    """MDI CLI."""
+    """Hafnia CLI."""
     ctx.obj = Config()
 
 
 @main.command("configure")
 @click.pass_obj
 def configure(cfg: Config) -> None:
-    """Configure MDI CLI settings."""
+    """Configure Hafnia CLI settings."""
 
     from hafnia.platform.api import get_organization_id
 
@@ -26,20 +26,16 @@ def configure(cfg: Config) -> None:
     except ValueError:
         raise click.ClickException(consts.ERROR_CREATE_PROFILE)
 
-    api_key = click.prompt("MDI API Key", type=str, hide_input=True)
+    api_key = click.prompt("Hafnia API Key", type=str, hide_input=True)
     try:
         cfg.api_key = api_key.strip()
     except ValueError as e:
         click.echo(f"Error: {str(e)}", err=True)
         return
-    platform_url = click.prompt(
-        "MDI Platform URL", type=str, default="https://api.mdi.milestonesys.com"
-    )
+    platform_url = click.prompt("Hafnia Platform URL", type=str, default="https://api.mdi.milestonesys.com")
     cfg.platform_url = platform_url.strip()
     try:
-        cfg.organization_id = get_organization_id(
-            cfg.get_platform_endpoint("organizations"), cfg.api_key
-        )
+        cfg.organization_id = get_organization_id(cfg.get_platform_endpoint("organizations"), cfg.api_key)
     except Exception:
         raise click.ClickException(consts.ERROR_ORG_ID)
     cfg.save_config()
@@ -51,7 +47,7 @@ def configure(cfg: Config) -> None:
 def clear(cfg: Config) -> None:
     """Remove stored configuration."""
     cfg.clear()
-    click.echo("Successfully cleared MDI configuration.")
+    click.echo("Successfully cleared Hafnia configuration.")
 
 
 main.add_command(profile_cmds.profile)

{hafnia-0.1.8 → hafnia-0.1.9}/src/cli/config.py

@@ -92,11 +92,9 @@ class Config:
         if config_env_path:
             return Path(config_env_path).expanduser()
 
-        return Path.home() / ".mdi" / "config.json"
+        return Path.home() / ".hafnia" / "config.json"
 
-    def add_profile(
-        self, profile_name: str, profile: ConfigSchema, set_active: bool = False
-    ) -> None:
+    def add_profile(self, profile_name: str, profile: ConfigSchema, set_active: bool = False) -> None:
         profile_name = profile_name.strip()
         self.config_data.profiles[profile_name] = profile
         if set_active:

{hafnia-0.1.8 → hafnia-0.1.9}/src/cli/consts.py

@@ -1,11 +1,7 @@
-ERROR_CONFIGURE: str = "Please configure the CLI with `mdi configure`"
-ERROR_PROFILE_NOT_EXIST: str = (
-    "No active profile configured. Please configure the CLI with `mdi configure`"
-)
-ERROR_PROFILE_REMOVE_ACTIVE: str = (
-    "Cannot remove active profile. Please switch to another profile first."
-)
-ERROR_API_KEY_NOT_SET: str = "API key not set. Please configure the CLI with `mdi configure`."
+ERROR_CONFIGURE: str = "Please configure the CLI with `hafnia configure`"
+ERROR_PROFILE_NOT_EXIST: str = "No active profile configured. Please configure the CLI with `hafnia configure`"
+ERROR_PROFILE_REMOVE_ACTIVE: str = "Cannot remove active profile. Please switch to another profile first."
+ERROR_API_KEY_NOT_SET: str = "API key not set. Please configure the CLI with `hafnia configure`."
 ERROR_ORG_ID: str = "Failed to fetch organization ID. Verify platform URL and API key."
 ERROR_CREATE_PROFILE: str = "Failed to create profile. Profile name must be unique and not empty."
 
@@ -15,4 +11,4 @@ ERROR_EXPERIMENT_DIR: str = "Source directory does not exist"
 
 PROFILE_SWITCHED_SUCCESS: str = "Switched to profile:"
 PROFILE_REMOVED_SUCCESS: str = "Removed profile:"
-PROFILE_TABLE_HEADER: str = "MDI Platform Profile:"
+PROFILE_TABLE_HEADER: str = "Hafnia Platform Profile:"

{hafnia-0.1.8 → hafnia-0.1.9}/src/cli/data_cmds.py

@@ -18,14 +18,12 @@ def data():
 @click.argument("destination")
 @click.pass_obj
 def data_get(cfg: Config, url: str, destination: click.Path) -> None:
-    """Download resource from MDI platform"""
+    """Download resource from Hafnia platform"""
 
     from hafnia.platform import download_resource
 
     try:
-        result = download_resource(
-            resource_url=url, destination=str(destination), api_key=cfg.api_key
-        )
+        result = download_resource(resource_url=url, destination=str(destination), api_key=cfg.api_key)
     except Exception:
         raise click.ClickException(consts.ERROR_GET_RESOURCE)
 
@@ -37,10 +35,8 @@ def data_get(cfg: Config, url: str, destination: click.Path) -> None:
 @click.argument("destination", default=None, required=False)
 @click.option("--force", is_flag=True, default=False, help="Force download")
 @click.pass_obj
-def data_download(
-    cfg: Config, dataset_name: str, destination: Optional[click.Path], force: bool
-) -> None:
-    """Download dataset from MDI platform"""
+def data_download(cfg: Config, dataset_name: str, destination: Optional[click.Path], force: bool) -> None:
+    """Download dataset from Hafnia platform"""
 
     from hafnia.data.factory import download_or_get_dataset_path
 
@@ -52,7 +48,7 @@ def data_download(
             endpoint=endpoint_dataset,
             api_key=api_key,
             output_dir=destination,
-            force=force,
+            force_redownload=force,
         )
     except Exception:
         raise click.ClickException(consts.ERROR_GET_RESOURCE)

{hafnia-0.1.8 → hafnia-0.1.9}/src/cli/experiment_cmds.py

@@ -20,9 +20,7 @@ def experiment() -> None:
 @click.argument("dataset_name")
 @click.argument("env_name")
 @click.pass_obj
-def create(
-    cfg: Config, name: str, source_dir: Path, exec_cmd: str, dataset_name: str, env_name: str
-) -> None:
+def create(cfg: Config, name: str, source_dir: Path, exec_cmd: str, dataset_name: str, env_name: str) -> None:
     """Create a new experiment run"""
     from hafnia.platform import (
         create_experiment,
@@ -35,25 +33,19 @@ def create(
         raise click.ClickException(consts.ERROR_EXPERIMENT_DIR)
 
     try:
-        dataset_id = get_dataset_id(
-            dataset_name, cfg.get_platform_endpoint("datasets"), cfg.api_key
-        )
+        dataset_id = get_dataset_id(dataset_name, cfg.get_platform_endpoint("datasets"), cfg.api_key)
     except (IndexError, KeyError):
         raise click.ClickException(f"Dataset '{dataset_name}' not found.")
     except Exception:
         raise click.ClickException(f"Error retrieving dataset '{dataset_name}'.")
 
     try:
-        recipe_id = create_recipe(
-            source_dir, cfg.get_platform_endpoint("recipes"), cfg.api_key, cfg.organization_id
-        )
+        recipe_id = create_recipe(source_dir, cfg.get_platform_endpoint("recipes"), cfg.api_key, cfg.organization_id)
     except Exception:
         raise click.ClickException(f"Failed to create recipe from '{source_dir}'")
 
     try:
-        env_id = get_exp_environment_id(
-            env_name, cfg.get_platform_endpoint("experiment_environments"), cfg.api_key
-        )
+        env_id = get_exp_environment_id(env_name, cfg.get_platform_endpoint("experiment_environments"), cfg.api_key)
     except Exception:
         raise click.ClickException(f"Environment '{env_name}' not found")
 

{hafnia-0.1.8 → hafnia-0.1.9}/src/cli/runc_cmds.py

@@ -28,9 +28,7 @@ def launch(task: str) -> None:
 @click.argument("ecr_repository", default="localhost")
 @click.argument("image_name", default="recipe")
 @click.pass_obj
-def build(
-    cfg: Config, recipe_url: str, state_file: str, ecr_repository: str, image_name: str
-) -> None:
+def build(cfg: Config, recipe_url: str, state_file: str, ecr_repository: str, image_name: str) -> None:
     """Build docker image with a given recipe."""
     from hafnia.platform.builder import build_image, prepare_recipe
 

{hafnia-0.1.8 → hafnia-0.1.9}/src/hafnia/data/factory.py

@@ -33,9 +33,7 @@ def download_or_get_dataset_path(
     dataset_path_sample = dataset_path_base / "sample"
 
     if dataset_path_sample.exists() and not force_redownload:
-        logger.info(
-            "Dataset found locally. Set 'force=True' or add `--force` flag with cli to re-download"
-        )
+        logger.info("Dataset found locally. Set 'force=True' or add `--force` flag with cli to re-download")
         return dataset_path_sample
 
     dataset_id = get_dataset_id(dataset_name, endpoint, api_key)
@@ -44,7 +42,7 @@ def download_or_get_dataset_path(
     if force_redownload and dataset_path_sample.exists():
         # Remove old files to avoid old files conflicting with new files
         shutil.rmtree(dataset_path_sample, ignore_errors=True)
-    status = download_resource(dataset_access_info_url, dataset_path_base, api_key)
+    status = download_resource(dataset_access_info_url, str(dataset_path_base), api_key)
     if status:
         return dataset_path_sample
     raise RuntimeError("Failed to download dataset")
@@ -68,7 +66,7 @@ def load_from_platform(
 
 
 def load_dataset(dataset_name: str, force_redownload: bool = False) -> Union[Dataset, DatasetDict]:
-    """Load a dataset either from a local path or from the MDI platform."""
+    """Load a dataset either from a local path or from the Hafnia platform."""
 
     if utils.is_remote_job():
         path_dataset = Path(os.getenv("MDI_DATASET_DIR", "/opt/ml/input/data/training"))

hafnia-0.1.9/src/hafnia/experiment/__init__.py

@@ -0,0 +1,3 @@
+from hafnia.experiment.hafnia_logger import HafniaLogger
+
+__all__ = ["HafniaLogger"]

hafnia-0.1.8/src/hafnia/experiment/mdi_logger.py → hafnia-0.1.9/src/hafnia/experiment/hafnia_logger.py

@@ -73,7 +73,7 @@ class Entity(BaseModel):
         )
 
 
-class MDILogger:
+class HafniaLogger:
     EXPERIMENT_FILE = "experiment.parquet"
 
     def __init__(self, log_dir: Union[Path, str] = "./.data"):

{hafnia-0.1.8 → hafnia-0.1.9}/src/hafnia/http.py

@@ -31,9 +31,7 @@ def fetch(endpoint: str, headers: Dict, params: Optional[Dict] = None) -> Dict:
         http.clear()
 
 
-def post(
-    endpoint: str, headers: Dict, data: Union[Path, Dict, bytes], multipart: bool = False
-) -> Dict:
+def post(endpoint: str, headers: Dict, data: Union[Path, Dict, bytes], multipart: bool = False) -> Dict:
     """Posts data to backend endpoint.
 
     Args:
@@ -76,9 +74,7 @@ def post(
 
         if response.status not in (200, 201):
             error_details = response.data.decode("utf-8")
-            raise urllib3.exceptions.HTTPError(
-                f"Request failed with status {response.status}: {error_details}"
-            )
+            raise urllib3.exceptions.HTTPError(f"Request failed with status {response.status}: {error_details}")
 
         return json.loads(response.data.decode("utf-8"))
     finally: