qwak-sdk 0.5.66__py3-none-any.whl → 0.5.67__py3-none-any.whl

qwak_sdk/__init__.py

@@ -1,6 +1,6 @@
 # fmt: off
 __author__ = '''Qwak.ai'''
-__version__ = '0.5.66'
+__version__ = '0.5.67'
 
 from qwak.inner import wire_dependencies
 

qwak_sdk/commands/models/build/ui.py

@@ -1,12 +1,21 @@
 import click
-from qwak.inner.build_config.build_config_v1 import ModelUri, RemoteBuildResources, BuildProperties, DockerConf, \
-    BuildConfigV1
+from qwak.inner.build_config.build_config_v1 import (
+    BuildConfigV1,
+    BuildProperties,
+    DockerConf,
+    ModelUri,
+    RemoteBuildResources,
+)
 from qwak.inner.build_logic.execute_build_pipeline import execute_build_pipeline
-from qwak.inner.build_logic.run_handlers.programmatic_phase_run_handler import ProgrammaticPhaseRunHandler
+from qwak.inner.build_logic.run_handlers.programmatic_phase_run_handler import (
+    ProgrammaticPhaseRunHandler,
+)
 from qwak.inner.const import QwakConstants
 
 from qwak_sdk.commands.models.build._logic.build_steps import create_pipeline
-from qwak_sdk.commands.models.build._logic.client_logs.cli_phase_run_handler import CLIPhaseRunHandler
+from qwak_sdk.commands.models.build._logic.client_logs.cli_phase_run_handler import (
+    CLIPhaseRunHandler,
+)
 from qwak_sdk.commands.models.build._logic.client_logs.logger import get_build_logger
 from qwak_sdk.commands.models.build._logic.client_logs.utils import zip_logs
 from qwak_sdk.inner.tools.cli_tools import QwakCommand
@@ -14,7 +23,12 @@ from qwak_sdk.inner.tools.config_handler import config_handler
 
 
 @click.command("build", cls=QwakCommand)
-@click.option("--model-id", metavar="NAME", required=False, help="Model ID")
+@click.option(
+    "--model-id",
+    metavar="NAME",
+    required=False,
+    help="Model ID to assign the build for",
+)
 @click.option(
     "--main-dir",
     metavar="NAME",
@@ -60,7 +74,7 @@ from qwak_sdk.inner.tools.config_handler import config_handler
     "--git-credentials-secret",
     metavar="NAME",
     required=False,
-    help="[REMOTE BUILD] Predefined Qwak secret secret name, that contains access credentials to private repositories"
+    help="Predefined Qwak secret secret name, that contains access credentials to private repositories"
     + "Secrets should be of the form USERNAME:ACCESS_TOKEN. For info regarding defining Qwak Secrets using the"
     + "`qwak secret` command",
 )
@@ -68,7 +82,7 @@ from qwak_sdk.inner.tools.config_handler import config_handler
     "--cpus",
     metavar="NAME",
     required=False,
-    help="[REMOTE BUILD] Number of cpus to use on the remote build. [Default (If GPU not configured): 2] "
+    help="Number of cpus to use on the remote build. [Default (If GPU not configured): 2] "
     "(DO NOT CONFIGURE GPU AND CPU TOGETHER)",
     type=click.FLOAT,
 )
@@ -76,14 +90,14 @@ from qwak_sdk.inner.tools.config_handler import config_handler
     "--memory",
     metavar="NAME",
     required=False,
-    help="[REMOTE BUILD] Memory to use on the remote build. [Default (If GPU not configured): 4Gi] "
+    help="Memory to use on the remote build. [Default (If GPU not configured): 4Gi] "
     "(DO NOT CONFIGURE GPU AND CPU TOGETHER)",
 )
 @click.option(
     "--gpu-type",
     metavar="NAME",
     required=False,
-    help=f"[REMOTE BUILD] Type of GPU to use on the remote build ({', '.join([x for x in QwakConstants.GPU_TYPES])})."
+    help=f"Type of GPU to use on the remote build ({', '.join([x for x in QwakConstants.GPU_TYPES])})."
     f"\n[Default: {RemoteBuildResources.gpu_type}]"
     "(DO NOT CONFIGURE GPU AND CPU TOGETHER)",
     type=click.STRING,
@@ -93,13 +107,13 @@ from qwak_sdk.inner.tools.config_handler import config_handler
     metavar="NAME",
     required=False,
     type=int,
-    help=f"[REMOTE BUILD] Amount of GPU to use on the remote build."
+    help=f"Amount of GPU's to use on the remote build."
     f"\n[Default: {RemoteBuildResources.gpu_amount}] "
     "(DO NOT CONFIGURE GPU AND CPU TOGETHER)",
 )
 @click.option(
     "--gpu-compatible",
-    help=f"[REMOTE BUILD] Whether to build an image that is compatible to be deployd on a GPU instance."
+    help=f"Whether to build an image that is compatible to be deployd on a GPU instance."
     f"\n[Default: {BuildProperties.gpu_compatible}] ",
     default=False,
     is_flag=True,
@@ -108,12 +122,12 @@ from qwak_sdk.inner.tools.config_handler import config_handler
     "--iam-role-arn",
     required=False,
     type=str,
-    help="[REMOTE BUILD] Custom IAM Role ARN.",
+    help="Custom IAM Role ARN for AWS based builds",
 )
 @click.option(
     "--service-account-key-secret-name",
     type=str,
-    help="Custom service account for Gcp.",
+    help="Custom service account for GCP",
 )
 @click.option(
     "--cache/--no-cache",
@@ -234,10 +248,15 @@ from qwak_sdk.inner.tools.config_handler import config_handler
 )
 @click.option(
     "--push/--no-push",
-    help="Whether to push the built image to the registry (default is True)",
+    help="Whether to push the build image to the registry (default is True)",
     default=True,
     is_flag=True,
 )
+@click.option(
+    "--provision-instance-timeout",
+    help="Timeout in minutes for the provision instance step",
+    default=120,
+)
 @click.argument("uri", required=False)
 def models_build(**kwargs):
     return build(**kwargs)
@@ -264,13 +283,17 @@ def build(
         id_only = kwargs.get("id_only")
         pipeline, success_msg = create_pipeline(config)
         with get_build_logger(config=config, json_logs=json_logs) as (
-                logger,
-                log_path,
+            logger,
+            log_path,
         ):
             if programmatic:
-                build_runner = ProgrammaticPhaseRunHandler(logger, config.verbose, json_logs)
+                build_runner = ProgrammaticPhaseRunHandler(
+                    logger, config.verbose, json_logs
+                )
             else:
-                build_runner = CLIPhaseRunHandler(logger, log_path, config.verbose, json_logs)
+                build_runner = CLIPhaseRunHandler(
+                    logger, log_path, config.verbose, json_logs
+                )
 
             execute_build_pipeline(
                 pipeline,

{qwak_sdk-0.5.66.dist-info → qwak_sdk-0.5.67.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: qwak-sdk
-Version: 0.5.66
+Version: 0.5.67
 Summary: Qwak SDK and CLI for qwak models
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model
@@ -37,8 +37,8 @@ Requires-Dist: pandas (>1.1.3) ; (python_version >= "3.9" and python_version < "
 Requires-Dist: pandas (>=1.4.0) ; (python_version >= "3.10" and python_version < "3.12") and (extra == "batch" or extra == "feedback")
 Requires-Dist: pyarrow (>=6.0.0,<11.0.0) ; extra == "batch"
 Requires-Dist: python-json-logger (>=2.0.2)
-Requires-Dist: qwak-core (==0.3.166)
-Requires-Dist: qwak-inference (>=0.1.7,<0.2.0)
+Requires-Dist: qwak-core (==0.4.8)
+Requires-Dist: qwak-inference (>=0.1.16,<0.2.0)
 Requires-Dist: rich (>=13.0.0)
 Requires-Dist: tabulate (>=0.8.0)
 Requires-Dist: yaspin (>=2.0.0)

{qwak_sdk-0.5.66.dist-info → qwak_sdk-0.5.67.dist-info}/RECORD

@@ -1,4 +1,4 @@
-qwak_sdk/__init__.py,sha256=Nuh00DOIPrs-L7xXJd0OURleVtzcyGG2Apfr4ghZtLU,135
+qwak_sdk/__init__.py,sha256=NNZZ1nm59YD8isM6cCdmZBmVnhSLrubQOw8xfd9VXpc,135
 qwak_sdk/cli.py,sha256=FIK1dUNxR57ypb-CeD7fKSJnPJ02lrjR9G4aj2qMLPU,2458
 qwak_sdk/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -121,7 +121,7 @@ qwak_sdk/commands/models/build/_logic/util/__init__.py,sha256=47DEQpj8HBSa-_TImW
 qwak_sdk/commands/models/build/_logic/util/protobuf_factory.py,sha256=ar_oY38w_x0sxgVF7EBs5h7gchNsDntvtKK5sSYxb24,1686
 qwak_sdk/commands/models/build/_logic/util/step_decorator.py,sha256=HLZyCGdqe3Ir7SaPWp1YNRHJpjXG-e-bbAvnOFysAVM,1913
 qwak_sdk/commands/models/build/_logic/util/text.py,sha256=tH-v19Mt8l90sMVxku5XRtrderT0qdRqJ-jLijqannA,188
-qwak_sdk/commands/models/build/ui.py,sha256=MX4OpBHh3X83JXc0FFsx4PP_4IZVVGe2bBS1Z5VK36o,8967
+qwak_sdk/commands/models/build/ui.py,sha256=qr-brINdA6wF9QVgFjE77qGi3KIYckHTrPQlr-d5jyo,9164
 qwak_sdk/commands/models/builds/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/models/builds/builds_commands_group.py,sha256=0nSfTY8TracXG61rFboQWUTXJisHO6dgtJKeijy6ru8,491
 qwak_sdk/commands/models/builds/cancel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -318,7 +318,7 @@ qwak_sdk/tools/colors.py,sha256=7pui_GGjC4uZKYFsIyXaJjYsjLxJVHb4OrfTgr93hqo,287
 qwak_sdk/tools/files.py,sha256=AyKJTOy7NhvP3SrqwIw_lxYNCOy1CvLgMmSJpWZ0OKM,2257
 qwak_sdk/tools/log_handling.py,sha256=Aa1EmxUPCX8YWiZRutUvnqPv6K_z1zoGMwIWsEv24mM,6327
 qwak_sdk/tools/utils.py,sha256=SHmU4r_m2ABZyFYMC03P17GvltPbYbmB39hvalIZEtI,1168
-qwak_sdk-0.5.66.dist-info/entry_points.txt,sha256=vSl0ELYDyj640oMM57u0AjBP87wtLYxCcGOendhEx80,47
-qwak_sdk-0.5.66.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
-qwak_sdk-0.5.66.dist-info/METADATA,sha256=12ofqmWZmT6yD_-9vCCrp7PwtF2vrOC5p7rz-yqejIU,2600
-qwak_sdk-0.5.66.dist-info/RECORD,,
+qwak_sdk-0.5.67.dist-info/entry_points.txt,sha256=vSl0ELYDyj640oMM57u0AjBP87wtLYxCcGOendhEx80,47
+qwak_sdk-0.5.67.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
+qwak_sdk-0.5.67.dist-info/METADATA,sha256=xfnTJi0rGKt4gDYgthZkCDE43GNuPOkdz8KYgS3hvEM,2599
+qwak_sdk-0.5.67.dist-info/RECORD,,