frogml-cli 0.0.1__tar.gz → 0.0.2__tar.gz

{frogml_cli-0.0.1 → frogml_cli-0.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-cli
-Version: 0.0.1
+Version: 0.0.2
 Summary: Frogml CLI for frogml models
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model,jfrog

{frogml_cli-0.0.1 → frogml_cli-0.0.2}/frogml_cli/__init__.py

@@ -1,6 +1,6 @@
 # fmt: off
 __author__ = '''JFrog Ltd.'''
-__version__ = '0.0.1'
+__version__ = '0.0.2'
 __name__ = 'frogml_cli'
 
 from frogml import wire_dependencies

{frogml_cli-0.0.1 → frogml_cli-0.0.2}/frogml_cli/commands/models/init/_logic/template/general/cookiecutter.json

@@ -2,5 +2,5 @@
     "model_directory": "model",
     "main_directory": "main",
     "test_directory": "tests",
-    "model_class_name": "modelPrediction"
+    "model_class_name": "ModelPrediction"
 }

{frogml_cli-0.0.1 → frogml_cli-0.0.2}/frogml_cli/commands/models/init/ui.py

@@ -1,3 +1,5 @@
+from typing import List
+
 import click
 from frogml.core.tools.logger.logger import get_frogml_logger
 
@@ -5,10 +7,14 @@ from frogml_cli.commands.models.init._logic.initialize_model_structure import (
     initialize_model_structure,
 )
 from frogml_cli.inner.tools.cli_tools import FrogMLCommand
+from frogml_cli.tools.utils import get_models_init_example_choices
 
 logger = get_frogml_logger()
 
 
+MODELS_INIT_EXAMPLE_CHOICES: List[str] = get_models_init_example_choices()
+
+
 @click.command("init", cls=FrogMLCommand)
 @click.option(
     "--model-directory", metavar="NAME", required=False, help="folder for model content"
@@ -23,10 +29,8 @@ logger = get_frogml_logger()
     "--example",
     metavar="NAME",
     required=False,
-    type=click.Choice(
-        ["titanic", "credit_risk", "churn", "titanic_poetry"], case_sensitive=True
-    ),
-    help="""Generate a fully functioning example of a Qwak based model. Options: titanic / credit_risk / churn""",
+    type=click.Choice(MODELS_INIT_EXAMPLE_CHOICES, case_sensitive=True),
+    help=f"Generate a fully functioning example of a Qwak based model. Options: {' / '.join(MODELS_INIT_EXAMPLE_CHOICES)}",
 )
 @click.argument("uri", metavar="URI", required=True)
 def model_init(

{frogml_cli-0.0.1 → frogml_cli-0.0.2}/frogml_cli/tools/utils.py

@@ -1,7 +1,8 @@
 import logging
 import sys
 from contextlib import contextmanager
-from typing import Callable, Optional
+from pathlib import Path
+from typing import Callable, List, Optional
 
 from frogml.core.tools.logger.logger import (
     get_frogml_logger,
@@ -43,3 +44,21 @@ def frogml_spinner(
             if end_text:
                 sp.text = end_text
             sp.ok("✅")
+
+
+def get_models_init_example_choices() -> List[str]:
+    """Dynamically get available template choices from the template directory."""
+    template_dir = (
+        Path(__file__).parent.parent
+        / "commands"
+        / "models"
+        / "init"
+        / "_logic"
+        / "template"
+    )
+
+    return [
+        item.name
+        for item in template_dir.iterdir()
+        if item.is_dir() and not item.name.startswith(".")
+    ]

{frogml_cli-0.0.1 → frogml_cli-0.0.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "frogml-cli"
-version = "0.0.1"
+version = "0.0.2"
 description = "Frogml CLI for frogml models"
 authors = ["JFrog Ltd. <contact@jfrog.com>"]
 readme = "README.md"