qwak-sdk 0.5.82__tar.gz → 0.5.84__tar.gz

{qwak_sdk-0.5.82 → qwak_sdk-0.5.84}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: qwak-sdk
-Version: 0.5.82
+Version: 0.5.84
 Summary: Qwak SDK and CLI for qwak models
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model
@@ -41,7 +41,7 @@ Requires-Dist: pyarrow (>=6.0.0,<11.0.0) ; extra == "batch" or extra == "feature
 Requires-Dist: pyathena (>=2.2.0,!=2.18.0) ; extra == "feature-store"
 Requires-Dist: pyspark (==3.4.2) ; extra == "feature-store"
 Requires-Dist: python-json-logger (>=2.0.2)
-Requires-Dist: qwak-core (==0.4.105)
+Requires-Dist: qwak-core (==0.4.111)
 Requires-Dist: qwak-inference (>=0.1.18,<0.2.0)
 Requires-Dist: rich (>=13.0.0)
 Requires-Dist: tabulate (>=0.8.0)

{qwak_sdk-0.5.82 → qwak_sdk-0.5.84}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "qwak-sdk"
-version = "0.5.82"
+version = "0.5.84"
 description = "Qwak SDK and CLI for qwak models"
 authors = ["Qwak <info@qwak.com>"]
 readme = "README.md"
@@ -29,7 +29,7 @@ qwak = "qwak_sdk.main:qwak_cli"
 
 [tool.poetry.dependencies]
 python = ">=3.7.1,<3.12"
-qwak-core = "==0.4.105"
+qwak-core = "==0.4.111"
 qwak-inference = "~0.1.18"
 croniter = "==1.4.1"
 tabulate = ">=0.8.0"

{qwak_sdk-0.5.82 → qwak_sdk-0.5.84}/qwak_sdk/__init__.py

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

{qwak_sdk-0.5.82 → qwak_sdk-0.5.84}/qwak_sdk/commands/feature_store/trigger/ui.py

@@ -1,4 +1,7 @@
 import click
+from _qwak_proto.qwak.feature_store.features.feature_set_service_pb2 import (
+    RunBatchFeatureSetResponse,
+)
 from qwak.clients.feature_store import FeatureRegistryClient
 from qwak.exceptions import QwakException
 
@@ -15,10 +18,12 @@ def trigger_feature_set(name, **kwargs):
     Trigger a batch feature set ingestion job
 
     Args:
-        name: feature set name that will be triggered
+        name: Feature set name that will be triggered.
     """
     try:
-        FeatureRegistryClient().run_feature_set(feature_set_name=name)
+        result: RunBatchFeatureSetResponse = FeatureRegistryClient().run_feature_set(
+            feature_set_name=name
+        )
     except Exception as e:
         print(
             f"{Color.RED} Failed to trigger a batch feature set ingestion for feature set {name} {Color.END}"
@@ -28,5 +33,7 @@ def trigger_feature_set(name, **kwargs):
         ) from e
 
     print(
-        f"{Color.GREEN}Successfully triggered a batch feature set ingestion for feature set {Color.YELLOW}{name}"
+        f"{Color.GREEN}Successfully triggered a batch feature set ingestion for feature set: {Color.YELLOW}{name}"
     )
+    if result.execution_id:
+        print(f"{Color.WHITE} Execution ID: {Color.YELLOW}{result.execution_id}")

{qwak_sdk-0.5.82 → qwak_sdk-0.5.84}/setup.py

@@ -134,7 +134,7 @@ install_requires = \
  'croniter==1.4.1',
  'gitpython>=2.1.0',
  'python-json-logger>=2.0.2',
- 'qwak-core==0.4.105',
+ 'qwak-core==0.4.111',
  'qwak-inference>=0.1.18,<0.2.0',
  'rich>=13.0.0',
  'tabulate>=0.8.0',
@@ -204,7 +204,7 @@ entry_points = \
 
 setup_kwargs = {
     'name': 'qwak-sdk',
-    'version': '0.5.82',
+    'version': '0.5.84',
     'description': 'Qwak SDK and CLI for qwak models',
     'long_description': '# Qwak SDK\n\nQwak is an end-to-end production ML platform designed to allow data scientists to build, deploy, and monitor their models in production with minimal engineering friction.\n',
     'author': 'Qwak',