qwak-core 0.4.272__py3-none-any.whl → 0.4.274__py3-none-any.whl

qwak_services_mock/mocks/batch_job_manager_service.py

@@ -36,6 +36,9 @@ from _qwak_proto.qwak.batch_job.v1.batch_job_service_pb2 import (
     StartWarmupJobRequest,
     StartWarmupJobResponse,
     TaskExecutionDetails,
+    UpdateTasksDetailsRequest,
+    UpdateTasksDetailsResponse,
+    BatchTaskDetails,
 )
 from _qwak_proto.qwak.batch_job.v1.batch_job_service_pb2_grpc import (
     BatchJobManagementServiceServicer,
@@ -51,6 +54,7 @@ class BatchJobManagerService(BatchJobManagementServiceServicer):
         self.id_to_batch_job: Dict[str, MockBatchJob] = dict()
         self.model_id_active_warmup: Set[str] = set()
         self.models_to_fail: Set[str] = set()
+        self.task_id_to_update_task_details: Dict[str, BatchTaskDetails] = dict()
 
     def StartBatchJob(
         self, request: StartBatchJobRequest, context
@@ -265,6 +269,25 @@ class BatchJobManagerService(BatchJobManagementServiceServicer):
         except Exception as e:
             raise_internal_grpc_error(context, e)
 
+    def UpdateTasksDetails(
+            self, request: UpdateTasksDetailsRequest, context
+    ) -> UpdateTasksDetailsResponse:
+        for task in request.tasks_details:
+            task_id = task.task_id
+            if not task_id :
+                raise ValueError("Task ID cannot be empty")
+            input_files = task.input_files_details
+
+            print(f"Updating task with ID: {task_id} and input files: {[file.path for file in input_files]}")
+            self.task_id_to_update_task_details[task_id] = task
+        return UpdateTasksDetailsResponse()
+
+    def get_task_details(self,task_id: str
+        ) -> Optional[BatchTaskDetails]:
+            """
+            Get task details by task ID
+            """
+            return self.task_id_to_update_task_details[task_id]
 
 class MockBatchJob:
     def __init__(self, request: StartBatchJobRequest):
@@ -331,4 +354,4 @@ def batch_job_to_message(batch_job: MockBatchJob) -> BatchJobMessage:
             for line in batch_job.report
         ],
         task_executions=batch_job.tasks,
-    )
+    )

qwak_core-0.4.272.dist-info/METADATA

@@ -1,53 +0,0 @@
-Metadata-Version: 2.3
-Name: qwak-core
-Version: 0.4.272
-Summary: Qwak Core contains the necessary objects and communication tools for using the Qwak Platform
-License: Apache-2.0
-Keywords: mlops,ml,deployment,serving,model
-Author: Qwak
-Author-email: info@qwak.com
-Requires-Python: >=3.9,<3.12
-Classifier: License :: OSI Approved :: Apache Software License
-Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.7
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: Implementation :: CPython
-Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Provides-Extra: feature-store
-Provides-Extra: local-core-dependencies
-Requires-Dist: PyYAML
-Requires-Dist: cachetools
-Requires-Dist: chevron (==0.14.0)
-Requires-Dist: cloudpickle (==2.2.1) ; extra == "feature-store"
-Requires-Dist: dacite (==1.8.1)
-Requires-Dist: dependency-injector (>=4.0)
-Requires-Dist: filelock
-Requires-Dist: frogml-storage (>=0.11.2)
-Requires-Dist: grpcio (>=1.57.0)
-Requires-Dist: grpcio-tools (>=1.47.0) ; (python_full_version >= "3.7.1" and python_version < "3.10") and (extra == "local-core-dependencies")
-Requires-Dist: grpcio-tools (>=1.56.2) ; (python_version >= "3.10") and (extra == "local-core-dependencies")
-Requires-Dist: joblib (>=1.3.2,<2.0.0)
-Requires-Dist: marshmallow-dataclass (>=8.5.8,<9.0.0)
-Requires-Dist: mypy-protobuf (>=3.0.0,<4.0.0) ; extra == "local-core-dependencies"
-Requires-Dist: protobuf (>=3.10,<4) ; python_full_version >= "3.7.1" and python_version < "3.10"
-Requires-Dist: protobuf (>=4.21.6) ; python_version >= "3.10"
-Requires-Dist: pyarrow (>=6.0.0) ; extra == "feature-store"
-Requires-Dist: pyathena (>=2.2.0,!=2.18.0) ; extra == "feature-store"
-Requires-Dist: pyspark (==3.4.2) ; extra == "feature-store"
-Requires-Dist: python-jose[cryptography] (>=3.4.0)
-Requires-Dist: python-json-logger (>=2.0.2)
-Requires-Dist: requests
-Requires-Dist: retrying (==1.3.4)
-Requires-Dist: typeguard (>=2,<3)
-Project-URL: Home page, https://www.qwak.com/
-Description-Content-Type: text/markdown
-
-# Qwak Core
-
-Qwak 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.
-Qwak Core contains all the objects and tools necessary to use the Qwak Platform
-