hafnia 0.5.1__py3-none-any.whl → 0.5.3__py3-none-any.whl

hafnia_cli/trainer_package_cmds.py

@@ -1,5 +1,5 @@
 from pathlib import Path
-from typing import Optional
+from typing import Dict, Optional
 
 import click
 
@@ -21,12 +21,60 @@ def cmd_list_trainer_packages(cfg: Config, limit: Optional[int]) -> None:
 
     from hafnia.platform.trainer_package import get_trainer_packages, pretty_print_trainer_packages
 
-    endpoint = cfg.get_platform_endpoint("trainers")
-    trainers = get_trainer_packages(endpoint, cfg.api_key)
+    trainers = get_trainer_packages(cfg=cfg)
 
     pretty_print_trainer_packages(trainers, limit=limit)
 
 
+@trainer_package.command(name="create")
+@click.pass_obj
+@click.argument(
+    "path",
+    type=Path,
+)
+@click.option(
+    "-n",
+    "--name",
+    type=str,
+    default=None,
+    help="Name of the trainer package.",
+)
+@click.option(
+    "-d",
+    "--description",
+    type=str,
+    default=None,
+    help="Description of the trainer package.",
+)
+@click.option(
+    "--cmd",
+    type=Optional[str],
+    default=None,
+    show_default=True,
+    help="Default command to run the trainer package.",
+)
+def cmd_create_trainer_package(
+    cfg: Config,
+    path: Path,
+    cmd: Optional[str] = None,
+    name: Optional[str] = None,
+    description: Optional[str] = None,
+) -> Dict:
+    """Create a trainer package on the platform"""
+    from hafnia.platform.trainer_package import create_trainer_package
+
+    path_trainer = Path(path).resolve()
+    trainer_response = create_trainer_package(
+        source_dir=path_trainer,
+        name=name,
+        description=description,
+        cmd=cmd,
+        cfg=cfg,
+    )
+
+    return trainer_response
+
+
 @trainer_package.command(name="create-zip")
 @click.argument("source")
 @click.option(
@@ -46,7 +94,7 @@ def cmd_create_trainer_package_zip(source: str, output: str) -> None:
         raise click.ClickException(consts.ERROR_TRAINER_PACKAGE_FILE_FORMAT)
 
     path_source = Path(source)
-    path_output_zip = archive_dir(path_source, path_output_zip)
+    path_output_zip, _ = archive_dir(path_source, path_output_zip)
 
 
 @trainer_package.command(name="view-zip")

hafnia/data/__init__.py

@@ -1,3 +0,0 @@
-from hafnia.data.factory import get_dataset_path, load_dataset
-
-__all__ = ["load_dataset", "get_dataset_path"]

hafnia/data/factory.py

@@ -1,22 +0,0 @@
-from pathlib import Path
-from typing import Any
-
-from hafnia import utils
-from hafnia.dataset.hafnia_dataset import HafniaDataset, get_or_create_dataset_path_from_recipe
-
-
-def load_dataset(recipe: Any, force_redownload: bool = False) -> HafniaDataset:
-    """Load a dataset either from a local path or from the Hafnia platform."""
-
-    path_dataset = get_dataset_path(recipe, force_redownload=force_redownload)
-    dataset = HafniaDataset.from_path(path_dataset)
-    return dataset
-
-
-def get_dataset_path(recipe: Any, force_redownload: bool = False) -> Path:
-    if utils.is_hafnia_cloud_job():
-        return utils.get_dataset_path_in_hafnia_cloud()
-
-    path_dataset = get_or_create_dataset_path_from_recipe(recipe, force_redownload=force_redownload)
-
-    return path_dataset