qwak-sdk 0.1.0__py3-none-any.whl → 0.2.20rc0__py3-none-any.whl

qwak_sdk/commands/models/executions/status/ui.py

@@ -0,0 +1,27 @@
+import click
+from qwak.exceptions import QwakException
+
+from qwak_sdk.commands.models.executions.status._logic import execute_execution_status
+from qwak_sdk.inner.tools.cli_tools import QwakCommand
+from qwak_sdk.inner.tools.logger.logger import get_qwak_logger
+
+logger = get_qwak_logger()
+
+
+@click.command(name="status", help="Get the status of an execution.", cls=QwakCommand)
+@click.option(
+    "--execution-id",
+    type=str,
+    help="The execution id.",
+)
+def execution_status(execution_id: str, **kwargs):
+    try:
+        job_status, success, fail_message = execute_execution_status(execution_id)
+        if not success:
+            raise QwakException(fail_message)
+
+        logger.info(f"Status is: {job_status}.")
+
+    except Exception as e:
+        logger.error(f"Failed to cancel execution, Error: {e}")
+        raise

qwak_sdk/commands/models/init/_logic/initialize_model_structure.py

@@ -0,0 +1,40 @@
+from pathlib import Path
+
+from cookiecutter import generate
+from qwak.exceptions import QwakException
+
+
+def initialize_model_structure(uri: str, template: str, logger, **template_args):
+    if Path(uri).is_file():
+        raise QwakException(
+            f"Unable to create the model directory structure - given path {uri} is an existing file"
+        )
+
+    if not Path(uri).is_dir():
+        logger.info(f"Given path {uri} doesn't exists - creating it")
+        Path(uri).mkdir(parents=True, exist_ok=True)
+
+    context = generate.generate_context(
+        context_file=(
+            Path(__file__).parent / "template" / template / "cookiecutter.json"
+        ).resolve(),
+        default_context=None,
+        extra_context={**template_args},
+    )
+
+    output_dir = uri + "/" + context.get("cookiecutter", {}).get("model_directory")
+    if Path(output_dir).exists():
+        raise QwakException(
+            f"Folder with the name {output_dir} is already exists. "
+            f"Please change model name or change the working path."
+        )
+    logger.info("Generating structure...")
+    generate.generate_files(
+        (Path(__file__).parent / "template" / template).resolve(),
+        context,
+        output_dir=uri,
+        overwrite_if_exists=False,
+        skip_if_file_exists=False,
+    )
+
+    logger.info(f"Created base model at {Path(output_dir).resolve()}")

qwak_sdk/commands/models/init/_logic/template/churn/cookiecutter.json

@@ -0,0 +1,3 @@
+{
+    "model_directory": "churn"
+}

qwak_sdk/commands/models/init/_logic/template/churn/{{cookiecutter.model_directory}}/main/__init__.py

@@ -0,0 +1,5 @@
+from .model import ChurnPrediction
+
+
+def load_model():
+    return ChurnPrediction()

qwak_sdk/commands/models/init/_logic/template/churn/{{cookiecutter.model_directory}}/main/conda.yml

@@ -0,0 +1,10 @@
+# name: churn-example
+channels:
+  - defaults
+  - conda-forge
+dependencies:
+  - python=3.8
+  - pip=22.2.2
+  - pandas=1.1.5
+  - scikit-learn=0.24
+  - catboost=0.24.3