hafnia 0.1.14__tar.gz → 0.1.16__tar.gz

{hafnia-0.1.14 → hafnia-0.1.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hafnia
-Version: 0.1.14
+Version: 0.1.16
 Summary: Python tools for communication with Hafnia platform.
 Author-email: Ivan Sahumbaiev <ivsa@milestone.dk>
 License-File: LICENSE
@@ -8,6 +8,7 @@ Requires-Python: >=3.10
 Requires-Dist: boto3>=1.35.91
 Requires-Dist: click>=8.1.8
 Requires-Dist: datasets>=3.2.0
+Requires-Dist: flatten-dict>=0.4.2
 Requires-Dist: pillow>=11.1.0
 Requires-Dist: pyarrow>=18.1.0
 Requires-Dist: pydantic>=2.10.4

{hafnia-0.1.14 → hafnia-0.1.16}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hafnia"
-version = "0.1.14"
+version = "0.1.16"
 description = "Python tools for communication with Hafnia platform."
 readme = "README.md"
 authors = [{ name = "Ivan Sahumbaiev", email = "ivsa@milestone.dk" }]
@@ -9,6 +9,7 @@ dependencies = [
     "boto3>=1.35.91",
     "click>=8.1.8",
     "datasets>=3.2.0",
+    "flatten-dict>=0.4.2",
     "pillow>=11.1.0",
     "pyarrow>=18.1.0",
     "pydantic>=2.10.4",

{hafnia-0.1.14 → hafnia-0.1.16}/src/cli/experiment_cmds.py

@@ -13,6 +13,32 @@ def experiment() -> None:
     pass
 
 
+@experiment.command(name="create_recipe")
+@click.option("--source_folder", default=".", type=Path, help="Path to the source folder", show_default=True)
+@click.option(
+    "--recipe_filename",
+    default="recipe.zip",
+    type=Path,
+    help="Recipe filename. Should have a '.zip' suffix",
+    show_default=True,
+)
+def create_recipe(source_folder: str, recipe_filename: str) -> None:
+    """Build recipe from local path as image with prefix - localhost"""
+
+    from hafnia.platform.builder import validate_recipe
+    from hafnia.utils import archive_dir
+
+    path_output_zip = Path(recipe_filename)
+
+    if path_output_zip.suffix != ".zip":
+        raise click.ClickException("Recipe filename must be a '.zip' file")
+
+    path_source = Path(source_folder)
+
+    path_output_zip = archive_dir(path_source, path_output_zip)
+    validate_recipe(path_output_zip)
+
+
 @experiment.command(name="create")
 @click.argument("name")
 @click.argument("source_dir", type=Path)

{hafnia-0.1.14 → hafnia-0.1.16}/src/hafnia/torch_helpers.py

@@ -3,7 +3,6 @@ from typing import List, Optional
 import datasets
 import torch
 import torchvision
-import torchvision.transforms.functional
 from flatten_dict import flatten
 from PIL import Image, ImageDraw, ImageFont
 from torchvision import tv_tensors

{hafnia-0.1.14 → hafnia-0.1.16}/uv.lock

@@ -549,12 +549,13 @@ http = [
 
 [[package]]
 name = "hafnia"
-version = "0.1.14"
+version = "0.1.16"
 source = { editable = "." }
 dependencies = [
     { name = "boto3" },
     { name = "click" },
     { name = "datasets" },
+    { name = "flatten-dict" },
     { name = "pillow" },
     { name = "pyarrow" },
     { name = "pydantic" },
@@ -585,6 +586,7 @@ requires-dist = [
     { name = "boto3", specifier = ">=1.35.91" },
     { name = "click", specifier = ">=8.1.8" },
     { name = "datasets", specifier = ">=3.2.0" },
+    { name = "flatten-dict", specifier = ">=0.4.2" },
     { name = "flatten-dict", marker = "extra == 'torch'", specifier = ">=0.4.2" },
     { name = "pillow", specifier = ">=11.1.0" },
     { name = "pyarrow", specifier = ">=18.1.0" },