qwak-sdk 0.5.24__py3-none-any.whl → 0.5.26__py3-none-any.whl

qwak_sdk/__init__.py

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

qwak_sdk/commands/feature_store/register/_logic.py

@@ -2,6 +2,7 @@ from typing import List, Optional, Tuple
 
 from _qwak_proto.qwak.feature_store.features.feature_set_pb2 import Feature
 from _qwak_proto.qwak.features_operator.v3.features_operator_pb2 import (
+    Outputs,
     SparkColumnDescription,
     ValidationSuccessResponse,
 )
@@ -129,6 +130,7 @@ def _handle_data_source_validation(
 
     response = getattr(result, result.WhichOneof("type"))
     if isinstance(response, ValidationSuccessResponse):
+        print_validation_outputs(response)
         print("✅ Validation completed successfully, got data source columns:")
 
         table = [
@@ -219,6 +221,7 @@ def _handle_batch_v1_feature_set_validation(
 
     response = getattr(result, result.WhichOneof("type"))
     if isinstance(response, ValidationSuccessResponse):
+        print_validation_outputs(response)
         print("✅ Validation completed successfully, got data source columns:")
         table = [
             (x.column_name, x.spark_type) for x in response.spark_column_description
@@ -288,3 +291,22 @@ def _batch_v1_feature_set_validation(
     else:
         print(f"Skipping validation for '{feature_set.name}' feature set")
     return spark_columns_features
+
+
+def print_validation_outputs(response: ValidationSuccessResponse) -> bool:
+    did_print = False
+    outputs = response.outputs
+
+    if isinstance(outputs, Outputs) and (outputs.stdout or outputs.stderr):
+        message = "Validation outputs: "
+
+        if outputs.stdout:
+            message += f"stdout: {outputs.stdout}\n "
+
+        if outputs.stderr:
+            message += f"stderr: {outputs.stderr}"
+
+        print(message)
+        did_print = True
+
+    return did_print

qwak_sdk/commands/models/build/_logic/config/config_v1.py

@@ -2,6 +2,7 @@ from dataclasses import dataclass, field
 from os.path import expandvars
 from typing import Dict, List, Optional, Tuple, Union
 
+from _qwak_proto.qwak.builds.build_pb2 import BaseDockerImageType
 from _qwak_proto.qwak.builds.builds_pb2 import (
     BuildConfiguration,
     BuildEnvironment,
@@ -16,10 +17,9 @@ from _qwak_proto.qwak.builds.builds_pb2 import (
     Step,
     VirtualEnvironmentBuild,
 )
+from _qwak_proto.qwak.instance_template.instance_template_pb2 import InstanceType
 from qwak.clients.build_orchestrator import BuildOrchestratorClient
 from qwak.clients.instance_template.client import InstanceTemplateManagementClient
-from _qwak_proto.qwak.builds.build_pb2 import BaseDockerImageType
-from _qwak_proto.qwak.instance_template.instance_template_pb2 import InstanceType
 from qwak.inner.tool.run_config import (
     QwakConfigBase,
     YamlConfigMixin,
@@ -32,6 +32,7 @@ from qwak.inner.tool.run_config import (
 
 from qwak_sdk.commands.models.build._logic.util.protobuf_factory import protobuf_factory
 from qwak_sdk.common.run_config import ConfigCliMap
+from qwak_sdk.common.run_config.utils import purchase_option_validation
 
 CONFIG_MAPPING: List[ConfigCliMap] = [
     # ConfigV1
@@ -97,6 +98,7 @@ CONFIG_MAPPING: List[ConfigCliMap] = [
     ConfigCliMap("verbose", "verbose", validate_int),
     ConfigCliMap("deploy", "deploy", validate_bool),
     ConfigCliMap("deployment_instance", "deployment_instance", validate_string),
+    ConfigCliMap("purchase_option", "purchase_option", purchase_option_validation),
 ]
 
 
@@ -217,6 +219,7 @@ class ConfigV1(YamlConfigMixin, QwakConfigBase):
     step: Step = field(default_factory=Step)
     deploy: bool = field(default=False)
     deployment_instance: Optional[str] = field(default=None)
+    purchase_option: Optional[str] = field(default=None)
     verbose: int = field(default=0)
 
     def __post_init__(self):
@@ -235,19 +238,33 @@ class ConfigV1(YamlConfigMixin, QwakConfigBase):
         self.build_env.docker.env_vars = list(self.build_env.docker.env_vars)
         self.build_env.docker.params = list(self.build_env.docker.params)
 
-    def fetch_base_docker_image_name(self, build_orchestrator: BuildOrchestratorClient, instance_template: InstanceTemplateManagementClient):
+    def fetch_base_docker_image_name(
+        self,
+        build_orchestrator: BuildOrchestratorClient,
+        instance_template: InstanceTemplateManagementClient,
+    ):
         if not self.build_env.docker.base_image:
-            is_gpu_used = self.build_env.remote.resources.gpu_type \
-                          and self.build_env.remote.resources.gpu_amount \
-                          and self.build_env.remote.resources.gpu_amount > 0
+            is_gpu_used = (
+                self.build_env.remote.resources.gpu_type
+                and self.build_env.remote.resources.gpu_amount
+                and self.build_env.remote.resources.gpu_amount > 0
+            )
 
             if self.build_env.remote.resources.instance and not is_gpu_used:
-                instance_spec = instance_template.get_instance_template(self.build_env.remote.resources.instance)
-                is_gpu_used = instance_spec.instance_type == InstanceType.INSTANCE_TYPE_GPU
+                instance_spec = instance_template.get_instance_template(
+                    self.build_env.remote.resources.instance
+                )
+                is_gpu_used = (
+                    instance_spec.instance_type == InstanceType.INSTANCE_TYPE_GPU
+                )
 
             if is_gpu_used or self.build_properties.gpu_compatible:
-                result = build_orchestrator.fetch_base_docker_image_name(BaseDockerImageType.GPU)
+                result = build_orchestrator.fetch_base_docker_image_name(
+                    BaseDockerImageType.GPU
+                )
                 self.build_env.docker.base_image = result.base_docker_image_name
             else:
-                result = build_orchestrator.fetch_base_docker_image_name(BaseDockerImageType.CPU)
+                result = build_orchestrator.fetch_base_docker_image_name(
+                    BaseDockerImageType.CPU
+                )
                 self.build_env.docker.base_image = result.base_docker_image_name

qwak_sdk/commands/models/build/ui.py

@@ -201,6 +201,13 @@ from qwak_sdk.inner.tools.config_handler import config_handler
     help="The instance size to deploy on - 'small', 'medium', 'large', etc...",
     default=None,
 )
+@click.option(
+    "--purchase-option",
+    required=False,
+    type=str,
+    help="The type of the instance to build on - 'ondemand' or 'spot'",
+    default=None,
+)
 @click.argument("uri", required=False)
 def models_build(**kwargs):
     return build(**kwargs)

qwak_sdk/commands/workspaces/update/_logic.py

@@ -12,7 +12,7 @@ from qwak_sdk.commands.workspaces.config.workspace_config import WorkspaceConfig
 
 def _update_workspace(config: WorkspaceConfig):
     """
-    Updating a new workspace
+    Updating a workspace
 
     Args:
         config: The workspace configuration

qwak_sdk/common/run_config/utils.py

@@ -198,6 +198,40 @@ def validate_variation(value: Any) -> bool:
     return True
 
 
+def purchase_option_validation(purchase_option: str, is_required: bool):
+    """Validate purchase option strings
+
+    Args:
+        purchase_option: the purchase option to validate.
+        is_required: True if value is required.
+
+    Returns:
+        bool: True if value is valid.
+
+    Examples:
+        >>> purchase_option_validation("spot", False)
+        True
+        >>> purchase_option_validation("ondemand", False)
+        True
+        >>> purchase_option_validation(None, False)
+        True
+        >>> purchase_option_validation("Ondemand", False)
+        False
+    """
+    if not purchase_option and not is_required:
+        return True
+    if purchase_option and (
+        not validate_string(purchase_option, False)
+        or purchase_option
+        not in [
+            "ondemand",
+            "spot",
+        ]
+    ):
+        return False
+    return True
+
+
 def validate_variations(value: Any, is_required: bool) -> bool:
     """Validate if value is a variation.
 

{qwak_sdk-0.5.24.dist-info → qwak_sdk-0.5.26.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: qwak-sdk
-Version: 0.5.24
+Version: 0.5.26
 Summary: Qwak SDK and CLI for qwak models
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model
@@ -27,7 +27,7 @@ Requires-Dist: pandas (<1.4) ; (python_full_version >= "3.7.1" and python_versio
 Requires-Dist: pandas (>=1.4.3,<2.0.0) ; (python_version >= "3.8" and python_version < "3.10") 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.0.212)
+Requires-Dist: qwak-core (==0.0.217)
 Requires-Dist: qwak-inference (==0.1.8)
 Requires-Dist: tabulate (>=0.8.0)
 Requires-Dist: yaspin (>=2.0.0)

{qwak_sdk-0.5.24.dist-info → qwak_sdk-0.5.26.dist-info}/RECORD

@@ -1,4 +1,4 @@
-qwak_sdk/__init__.py,sha256=piQZEtKNuGjuodFyerNkMl0e7vC_UHb5Eha_YOpRlrw,135
+qwak_sdk/__init__.py,sha256=_sJYnhQp9CgmuCiUM6pYmC2oumKITETcHT0VqbeM-E8,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
@@ -87,7 +87,7 @@ qwak_sdk/commands/feature_store/list/ui.py,sha256=NuXnQ3cdXSDCFAcC2jmgx4sAdjNuyI
 qwak_sdk/commands/feature_store/pause/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/feature_store/pause/ui.py,sha256=C89yl4ly-RpyZub30q5xV5g1UDnI_GmHSndvXAkXkw4,695
 qwak_sdk/commands/feature_store/register/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qwak_sdk/commands/feature_store/register/_logic.py,sha256=saR4ldU-pp-GEH3rVhgmALBvK3ywhtFZ6YWYZOIRh_o,10893
+qwak_sdk/commands/feature_store/register/_logic.py,sha256=TYXaa1H4kn1bf5vgGJYtIlBx_arJQoZXLm2wlDmMF88,11472
 qwak_sdk/commands/feature_store/register/ui.py,sha256=WDtQ1V0WoPDir14mZ_N8_RDFZfe68QrtqLDlf64eNwY,2822
 qwak_sdk/commands/feature_store/resume/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/feature_store/resume/ui.py,sha256=nI87xvA30qNQVJnT67lYJgwKGBtvZAurC6XX9ttpCZA,700
@@ -108,7 +108,7 @@ qwak_sdk/commands/models/build/_logic/client_logs/notifier_impl.py,sha256=TLAvkm
 qwak_sdk/commands/models/build/_logic/client_logs/spinner.py,sha256=iph1eVC7QtSzNobNP2j_x0DC2k7bbka9eT8rcBpZGeQ,359
 qwak_sdk/commands/models/build/_logic/client_logs/time_source.py,sha256=UFio1BSAffLLWZh617nIj7YjwH2IUI9HkWheqrn9UxE,975
 qwak_sdk/commands/models/build/_logic/client_logs/utils.py,sha256=0VcMlmWHtOR1fR4JUDQTDqeQmi3tH3cG8pBZf5Hfusc,311
-qwak_sdk/commands/models/build/_logic/config/config_v1.py,sha256=cSbH33dV3TedtBIouFFHOMCkmKmsQr4PigzuhbxbjT8,10523
+qwak_sdk/commands/models/build/_logic/config/config_v1.py,sha256=BJPzm5CaIg_xKioncUxxFkpvcno0KOHj2jAwZ7_krWs,10929
 qwak_sdk/commands/models/build/_logic/constant/host_resource.py,sha256=b3Bibo2v7bfDLYsMJsQJrkWQxOVMCboxuuvwBN-C8OI,131
 qwak_sdk/commands/models/build/_logic/constant/step_description.py,sha256=hrzWSNAjI_I_9iX3FTOIA_qjWDvWrhNjO3bueg1iaEY,881
 qwak_sdk/commands/models/build/_logic/constant/temp_dir.py,sha256=_tms51gfEZFGuQE4GmXOApi9OW7GeXPV_NBRVx2azkQ,73
@@ -145,7 +145,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=fSEn7_oVWcp-9R_h5JmFlqQMJjV5-ZUQgVzg3pDdoqs,1181
 qwak_sdk/commands/models/build/_logic/util/text.py,sha256=tH-v19Mt8l90sMVxku5XRtrderT0qdRqJ-jLijqannA,188
-qwak_sdk/commands/models/build/ui.py,sha256=z5lElqAy39Al1ozkKFJ4qYkjW6uJg1eGHOuQ8mYw9sg,6818
+qwak_sdk/commands/models/build/ui.py,sha256=9R0QCAzpvmffDmFPKZ8uNGmflRaziN7QzAD3IG1Hvuc,6984
 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
@@ -325,13 +325,13 @@ qwak_sdk/commands/workspaces/stop/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeR
 qwak_sdk/commands/workspaces/stop/_logic.py,sha256=3_dqIr_ae-GQvauykoIXlgBKl8roNesMrP9L8PJy3v4,1257
 qwak_sdk/commands/workspaces/stop/ui.py,sha256=XhuBoU7x1MrEDQcfQuxiuLEb2yQvbrDzlbcDBJ72nvk,599
 qwak_sdk/commands/workspaces/update/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qwak_sdk/commands/workspaces/update/_logic.py,sha256=tSxMOAGhTU0D9C-lbd49Ah7rEtf9eoEMeUXU9--gWMQ,2650
+qwak_sdk/commands/workspaces/update/_logic.py,sha256=Xx-Fzqc95FeSEsE0e3hWjfeWriDBCPmaKPKLlJIfEAM,2646
 qwak_sdk/commands/workspaces/update/ui.py,sha256=pqS4SosS145ciZoIkFhW951Y_Ec86PKIldCiv3iEU8w,1515
 qwak_sdk/commands/workspaces/workspaces_commands_group.py,sha256=yuW-UC73J_vp-i4fILX90TAUrxaFocAmltBrRIg67lY,795
 qwak_sdk/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/common/run_config/__init__.py,sha256=E73GRxqdEbTYsdd5EE7Wu4YXxTjxF4YkrBY_7-hZ4dU,437
 qwak_sdk/common/run_config/base.py,sha256=lbCl2DXBM90IXyLCFR2dUY35afwJlp9D4KgM8HP5_MQ,3166
-qwak_sdk/common/run_config/utils.py,sha256=g7HoOapU5Rog3lxZGmc5ZkiETmgmKN4YW0AJ7tdaiWQ,6087
+qwak_sdk/common/run_config/utils.py,sha256=2WvNxPiNhdqXBK5FzDMWl4ersdD7rX7KSecc4LMRLZc,6960
 qwak_sdk/exceptions/__init__.py,sha256=wWO9G6LyuA2zP_vbH3DbeHKUNSoZw9xvWqQL-Bt9MhI,381
 qwak_sdk/exceptions/qwak_command_exception.py,sha256=R2gZ12PA-_32UcxbC19k6a0JJ1QVywjAfh8gMS3p2fE,48
 qwak_sdk/exceptions/qwak_deploy_new_build_failed.py,sha256=KhGraVwBYK8xE2ObPW-LMWjmBVPaaG4BTZETxxDyU_Y,133
@@ -355,7 +355,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=CC9XC3z3eHr_fehQgDsXZr4VP29VaMRwC05xvC5Wue4,5616
 qwak_sdk/tools/utils.py,sha256=SHmU4r_m2ABZyFYMC03P17GvltPbYbmB39hvalIZEtI,1168
-qwak_sdk-0.5.24.dist-info/entry_points.txt,sha256=vSl0ELYDyj640oMM57u0AjBP87wtLYxCcGOendhEx80,47
-qwak_sdk-0.5.24.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
-qwak_sdk-0.5.24.dist-info/METADATA,sha256=xgkhDuuwxYPcpkoQmBJ9NG4oCyqPlposFOM1vCudhKc,1852
-qwak_sdk-0.5.24.dist-info/RECORD,,
+qwak_sdk-0.5.26.dist-info/entry_points.txt,sha256=vSl0ELYDyj640oMM57u0AjBP87wtLYxCcGOendhEx80,47
+qwak_sdk-0.5.26.dist-info/WHEEL,sha256=vVCvjcmxuUltf8cYhJ0sJMRDLr1XsPuxEId8YDzbyCY,88
+qwak_sdk-0.5.26.dist-info/METADATA,sha256=5i_ZdGveFUyeilkY7TbGbH-_zOmZijcYb2pKJxOBcPk,1852
+qwak_sdk-0.5.26.dist-info/RECORD,,