fiddler-client 3.0.3__tar.gz → 3.0.4__tar.gz

{fiddler_client-3.0.3/fiddler_client.egg-info → fiddler_client-3.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fiddler-client
-Version: 3.0.3
+Version: 3.0.4
 Summary: Python client for Fiddler Platform
 Home-page: https://fiddler.ai
 Author: Fiddler Labs

fiddler_client-3.0.4/fiddler/VERSION

@@ -0,0 +1 @@
+3.0.4

{fiddler_client-3.0.3 → fiddler_client-3.0.4}/fiddler/entities/model_deployment.py

@@ -6,7 +6,9 @@ from uuid import UUID
 
 from fiddler.decorators import handle_api_error
 from fiddler.entities.base import BaseEntity
+from fiddler.entities.job import Job
 from fiddler.entities.project import ProjectCompactMixin
+from fiddler.schemas.job import JobCompactResp
 from fiddler.schemas.model_deployment import ModelDeploymentResponse
 from fiddler.utils.logger import get_logger
 
@@ -94,7 +96,7 @@ class ModelDeployment(
         self._resp = resp_obj
 
     @handle_api_error
-    def update(self) -> None:
+    def update(self) -> Job:
         """Update an existing model deployment."""
         payload: dict[str, Any] = {'active': self.active}
         if self.replicas is not None:
@@ -108,6 +110,8 @@ class ModelDeployment(
             url=self._get_url(model_id=self.model_id), data=payload
         )
         self._refresh_from_response(response=response)
+        job_compact = JobCompactResp(**response.json()['data']['job'])
+        return Job.get(id_=job_compact.id)
 
     @classmethod
     def of(cls, model_id: UUID | str) -> ModelDeployment:

{fiddler_client-3.0.3 → fiddler_client-3.0.4}/fiddler/tests/apis/test_model_deployment.py

@@ -3,9 +3,12 @@ from http import HTTPStatus
 import pytest
 import responses
 
+from fiddler.entities.job import Job
 from fiddler.entities.model_deployment import ModelDeployment
 from fiddler.exceptions import NotFound
 from fiddler.tests.constants import (
+    JOB_ID,
+    JOB_NAME,
     MODEL_DEPLOYMENT_ID,
     MODEL_ID,
     MODEL_NAME,
@@ -50,6 +53,7 @@ API_RESPONSE_200 = {
             'full_name': USER_NAME,
             'email': USER_EMAIL,
         },
+        'job': {'id': JOB_ID, 'name': 'Deploy artifact model - manual'},
     },
 }
 
@@ -67,6 +71,31 @@ API_RESPONSE_404 = {
     }
 }
 
+JOB_API_RESPONSE_200 = {
+    'api_version': '3.0',
+    'kind': 'NORMAL',
+    'data': {
+        'name': JOB_NAME,
+        'info': {
+            'resource_type': 'MODEL',
+            'resource_name': 'bank_churn',
+            'project_name': 'bank_churn',
+        },
+        'id': JOB_ID,
+        'status': 'SUCCESS',
+        'progress': 100,
+        'error_message': None,
+        'error_reason': None,
+        'extras': {
+            'e36d1cf2-766f-4705-8269-b6f93bf1ca14': {
+                'status': 'SUCCESS',
+                'result': {'result': 'Success'},
+                'error_message': None,
+            }
+        },
+    },
+}
+
 
 @responses.activate
 def test_update_model_deployment_success() -> None:
@@ -79,8 +108,13 @@ def test_update_model_deployment_success() -> None:
     model_deployment.cpu = 300
     model_deployment.active = True
 
-    model_deployment.update()
+    responses.get(
+        url=f'{URL}/v3/jobs/{JOB_ID}',
+        json=JOB_API_RESPONSE_200,
+    )
+    job = model_deployment.update()
     assert isinstance(model_deployment, ModelDeployment)
+    assert isinstance(job, Job)
     assert model_deployment.cpu == 300
     assert model_deployment.active
 

{fiddler_client-3.0.3 → fiddler_client-3.0.4/fiddler_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fiddler-client
-Version: 3.0.3
+Version: 3.0.4
 Summary: Python client for Fiddler Platform
 Home-page: https://fiddler.ai
 Author: Fiddler Labs

fiddler_client-3.0.3/fiddler/VERSION

@@ -1 +0,0 @@
-3.0.3