fiddler-client 3.1.0.dev3__tar.gz → 3.1.0.dev4__tar.gz

{fiddler-client-3.1.0.dev3/fiddler_client.egg-info → fiddler_client-3.1.0.dev4}/PKG-INFO

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

fiddler_client-3.1.0.dev4/fiddler/VERSION

@@ -0,0 +1 @@
+3.1.0.dev4

{fiddler-client-3.1.0.dev3 → fiddler_client-3.1.0.dev4}/fiddler/entities/alert_rule.py

@@ -8,8 +8,8 @@ from uuid import UUID
 from fiddler.constants.alert_rule import AlertCondition, BinSize, CompareTo, Priority
 from fiddler.decorators import handle_api_error
 from fiddler.entities.base import BaseEntity
-from fiddler.entities.baseline import Baseline, BaselineCompactMixin
-from fiddler.entities.model import Model, ModelCompactMixin
+from fiddler.entities.baseline import BaselineCompactMixin
+from fiddler.entities.model import ModelCompactMixin
 from fiddler.entities.project import ProjectCompactMixin
 from fiddler.schemas.alert_rule import AlertRuleResp, NotificationConfig
 from fiddler.schemas.filter_query import OperatorType, QueryCondition, QueryRule
@@ -63,6 +63,12 @@ class AlertRule(
 
     @staticmethod
     def _get_url(id_: UUID | str | None = None) -> str:
+        """Get model resource/item url."""
+        url = '/v3/alert-rules'
+        return url if not id_ else f'{url}/{id_}'
+
+    @staticmethod
+    def _get_url_v2(id_: UUID | str | None = None) -> str:
         """Get model resource/item url."""
         url = '/v2/alert-configs'
         return url if not id_ else f'{url}/{id_}'
@@ -78,7 +84,7 @@ class AlertRule(
         instance = cls(
             name=resp_obj.name,
             model_id=resp_obj.model.id,
-            metric_id=resp_obj.metric_id,
+            metric_id=resp_obj.metric.id,
             priority=resp_obj.priority,
             compare_to=resp_obj.compare_to,
             condition=resp_obj.condition,
@@ -113,11 +119,11 @@ class AlertRule(
         # Reset properties
         self.model_id = resp_obj.model.id
         self.project_id = resp_obj.project.id
+        self.metric_id = resp_obj.metric.id
 
         # Add remaining fields
         fields = [
             'id',
-            'metric_id',
             'created_at',
             'updated_at',
         ]
@@ -179,7 +185,9 @@ class AlertRule(
         if columns:
             for column in columns:
                 rules.append(
-                    QueryRule(field='columns', operator=OperatorType.ANY, value=column)
+                    QueryRule(
+                        field='feature_names', operator=OperatorType.ANY, value=column
+                    )
                 )
 
         _filter = QueryCondition(rules=rules)
@@ -188,7 +196,6 @@ class AlertRule(
         if ordering:
             params['ordering'] = ','.join(ordering)
 
-        params['organization_name'] = cls.get_organization_name()
         for rule in cls._paginate(url=cls._get_url(), params=params):
             yield cls._from_dict(data=rule)
 
@@ -202,12 +209,9 @@ class AlertRule(
     @handle_api_error
     def create(self) -> AlertRule:
         """Create a new alert rule."""
-        model = Model.get(self.model_id)
         payload: dict[str, Any] = {
             'name': self.name,
-            'organization_name': self.organization_name,
-            'model_name': model.name,
-            'project_name': model.project.name,
+            'model_id': self.model_id,
             'metric_id': self.metric_id,
             'priority': self.priority,
             'compare_to': self.compare_to,
@@ -218,11 +222,9 @@ class AlertRule(
             'warning_threshold': self.warning_threshold,
             'feature_names': self.columns,
             'compare_bin_delta': self.compare_bin_delta,
-            'notifications': self._get_notifications_dict(),
         }
         if self.baseline_id:
-            baseline = Baseline.get(self.baseline_id)
-            payload['baseline_name'] = baseline.name
+            payload['baseline_id'] = self.baseline_id
 
         response = self._client().post(
             url=self._get_url(),
@@ -236,7 +238,7 @@ class AlertRule(
     def enable_notifications(self) -> None:
         """Enable notifications for an alert rule"""
         self._client().patch(
-            url=self._get_url(id_=self.id), data={'enable_notification': True}
+            url=self._get_url_v2(id_=self.id), data={'enable_notification': True}
         )
         logger.info(
             'Notifications have been enabled for alert rule with id: %s', self.id
@@ -246,7 +248,7 @@ class AlertRule(
     def disable_notifications(self) -> None:
         """Disable notifications for an alert rule"""
         self._client().patch(
-            url=self._get_url(id_=self.id), data={'enable_notification': False}
+            url=self._get_url_v2(id_=self.id), data={'enable_notification': False}
         )
         logger.info(
             'Notifications have been disabled for alert rule with id: %s', self.id
@@ -277,7 +279,7 @@ class AlertRule(
             webhooks=webhooks,
         )
         response = self._client().patch(
-            url=self._get_url(id_=self.id),
+            url=self._get_url_v2(id_=self.id),
             data={'notifications': notifications},
         )
 
@@ -293,7 +295,7 @@ class AlertRule(
         :return: NotificationConfig object
         """
 
-        response = self._client().get(url=self._get_url(id_=self.id))
+        response = self._client().get(url=self._get_url_v2(id_=self.id))
 
         return self._get_notifications_from_dict(
             response.json()['data']['notifications']

{fiddler-client-3.1.0.dev3 → fiddler_client-3.1.0.dev4}/fiddler/entities/model.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import builtins
 import typing
+from copy import deepcopy
 from dataclasses import dataclass
 from datetime import datetime
 from functools import cached_property
@@ -39,7 +40,6 @@ if typing.TYPE_CHECKING:
     from fiddler.entities.baseline import Baseline
     from fiddler.entities.dataset import Dataset
 
-
 logger = get_logger(__name__)
 
 
@@ -327,17 +327,17 @@ class Model(
         return Model(
             name=self.name,
             project_id=self.project_id,
-            schema=self.schema,
-            spec=self.spec,
+            schema=deepcopy(self.schema),
+            spec=deepcopy(self.spec),
             version=version if version else self.version,
             input_type=self.input_type,
             task=self.task,
-            task_params=self.task_params,
+            task_params=deepcopy(self.task_params),
             description=self.description,
             event_id_col=self.event_id_col,
             event_ts_col=self.event_ts_col,
             event_ts_format=self.event_ts_format,
-            xai_params=self.xai_params,
+            xai_params=deepcopy(self.xai_params),
         )
 
     @property

{fiddler-client-3.1.0.dev3 → fiddler_client-3.1.0.dev4}/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.1.0.dev3 → fiddler_client-3.1.0.dev4}/fiddler/packtools/template_model.py

@@ -36,7 +36,8 @@ class TemplateKerasTF2Model:
         :param hybrid: Boolean value for hybrid data type (for example matrices). Default to False.
         """
         import tensorflow as tf
-        from packtools.keras_ig_helpers import ComputeGradientsKerasTF2
+
+        from fiddler.packtools.keras_ig_helpers import ComputeGradientsKerasTF2
 
         embedding_names = embedding_names or []
         batch_input_shape_list = batch_input_shape_list or []
@@ -197,7 +198,7 @@ class TemplateTreeShap:
         :return: explanations_by_output and extras_by_output: two dictionaries with keys the output columns and values
                  the corresponding explanations and extra information necessary
         """
-        from packtools.project_attributions_helpers import TreeShapAttributions
+        from fiddler.packtools.project_attributions_helpers import TreeShapAttributions
 
         if explanation_name != self.tree_shap_explanations:
             raise NotImplementedError(

{fiddler-client-3.1.0.dev3 → fiddler_client-3.1.0.dev4}/fiddler/schemas/alert_rule.py

@@ -12,25 +12,33 @@ from fiddler.schemas.model import ModelCompactResp
 from fiddler.schemas.project import ProjectCompactResp
 
 
+class MetricResp(BaseModel):
+    id: Union[str, UUID]
+    display_name: str
+    type: str
+    type_display_name: str
+
+
 class AlertRuleResp(BaseModel):
-    id: UUID = Field(alias='uuid')
+    id: UUID
     name: str
-    model: ModelCompactResp
-    project: ProjectCompactResp
-    baseline: Optional[BaselineCompactResp]
-    segment: Optional[SegmentCompactResp]
     priority: Union[str, Priority]
     compare_to: Union[str, CompareTo]
-    metric_id: Union[str, UUID]
-    critical_threshold: float
     condition: Union[str, AlertCondition]
+    compare_bin_delta: Optional[int]
     bin_size: Union[str, BinSize]
     columns: Optional[List[str]] = Field(alias='feature_names')
-    compare_bin_delta: Optional[int]
+    critical_threshold: float
     warning_threshold: Optional[float]
 
     created_at: datetime
-    updated_at: datetime = Field(alias='last_updated')
+    updated_at: datetime
+
+    metric: MetricResp
+    model: ModelCompactResp
+    project: ProjectCompactResp
+    baseline: Optional[BaselineCompactResp]
+    segment: Optional[SegmentCompactResp]
 
 
 class NotificationConfig(BaseModel):

{fiddler-client-3.1.0.dev3 → fiddler_client-3.1.0.dev4}/fiddler/tests/apis/test_alert_rule.py

@@ -27,6 +27,51 @@ from fiddler.tests.constants import (
 from fiddler.utils.logger import set_logging
 
 API_RESPONSE_200 = {
+    'data': {
+        'id': ALERT_RULE_ID,
+        'model': {'id': MODEL_ID, 'name': MODEL_NAME, 'version': 'v1'},
+        'project': {'id': PROJECT_ID, 'name': PROJECT_NAME},
+        'organization': {
+            'id': 'e03d6f02-6efa-4eb1-b828-7ea9d2a083d0',
+            'name': 'scale154',
+        },
+        'baseline': None,
+        'segment': None,
+        'version': 'rule_v3',
+        'name': 'skdfvbj',
+        'priority': 'HIGH',
+        'feature_names': ['gender'],
+        'bin_size': 'Hour',
+        'compare_to': 'raw_value',
+        'condition': 'greater',
+        'warning_threshold': 1.0,
+        'critical_threshold': 2.0,
+        'updated_at': '2024-04-17T12:42:15.672909+00:00',
+        'enable_notification': True,
+        'compare_bin_delta': 0,
+        'created_at': '2024-04-17T12:42:15.672909+00:00',
+        'created_by': {
+            'id': '287949c2-13f8-49fc-a559-55b2e6f36e9b',
+            'full_name': 'Fiddler Administrator',
+            'email': 'admin@fiddler.ai',
+        },
+        'updated_by': {
+            'id': '287949c2-13f8-49fc-a559-55b2e6f36e9b',
+            'full_name': 'Fiddler Administrator',
+            'email': 'admin@fiddler.ai',
+        },
+        'metric': {
+            'id': 'null_violation_percentage',
+            'display_name': '% Missing Value Violation',
+            'type': 'data_integrity',
+            'type_display_name': 'Data Integrity',
+        },
+    },
+    'api_version': '3.0',
+    'kind': 'NORMAL',
+}
+
+API_RESPONSE_200_V2 = {
     'data': {
         'id': 794,
         'organization_name': 'mainbuild',
@@ -89,6 +134,7 @@ API_RESPONSE_200 = {
     'kind': 'NORMAL',
 }
 
+
 API_RESPONSE_404 = {
     'error': {
         'code': 404,
@@ -116,46 +162,44 @@ LIST_API_RESPONSE = {
         'items': [
             API_RESPONSE_200['data'],
             {
-                'critical_threshold': 100.0,
-                'created_by': 'admin@fiddler.ai',
-                'metric_display_name': 'Traffic',
-                'condition': 'lesser',
-                'metric': 'traffic',
-                'metric_id': 'traffic',
+                'id': '3431b746-d18f-4aed-99ea-2da8d5b46fb6',
+                'model': {'id': MODEL_ID, 'name': MODEL_NAME, 'version': 'v1'},
+                'project': {'id': PROJECT_ID, 'name': PROJECT_NAME},
+                'organization': {
+                    'id': 'e03d6f02-6efa-4eb1-b828-7ea9d2a083d0',
+                    'name': 'scale154',
+                },
+                'baseline': {'id': BASELINE_ID, 'name': BASELINE_NAME},
+                'segment': None,
+                'version': 'rule_v3',
+                'name': 'sfdvsefv',
+                'priority': 'HIGH',
+                'feature_names': ['numofproducts'],
                 'bin_size': 'Hour',
+                'compare_to': 'raw_value',
+                'condition': 'greater',
+                'warning_threshold': 1e-05,
+                'critical_threshold': 0.001,
+                'updated_at': '2024-04-17T12:43:23.871997+00:00',
                 'enable_notification': True,
-                'is_active': True,
-                'project_name': PROJECT_NAME,
-                'project': {
-                    'id': PROJECT_ID,
-                    'name': PROJECT_NAME,
+                'compare_bin_delta': 0,
+                'created_at': '2024-04-17T12:43:23.871997+00:00',
+                'created_by': {
+                    'id': '287949c2-13f8-49fc-a559-55b2e6f36e9b',
+                    'full_name': 'Fiddler Administrator',
+                    'email': 'admin@fiddler.ai',
                 },
-                'name': 'traffic-lt-1hr',
-                'created_at': '2023-11-28T08:04:10.916498+00:00',
-                'baseline_name': 'DEFAULT (bank_churn)',
-                'baseline': {
-                    'id': BASELINE_ID,
-                    'name': BASELINE_NAME,
+                'updated_by': {
+                    'id': '287949c2-13f8-49fc-a559-55b2e6f36e9b',
+                    'full_name': 'Fiddler Administrator',
+                    'email': 'admin@fiddler.ai',
                 },
-                'organization_name': 'mainbuild',
-                'priority': 'HIGH',
-                'category': None,
-                'warning_threshold': 200.0,
-                'segment': None,
-                'alert_type': 'service_metrics',
-                'time_bucket': 3600000,
-                'alert_type_display_name': 'Traffic',
-                'compare_period': 5,
-                'compare_to': 'time_period',
-                'uuid': '42eb6d51-dc5c-4e04-82aa-440b2d840704',
-                'id': 618,
-                'feature_names': None,
-                'model_name': MODEL_NAME,
-                'model': {
-                    'id': MODEL_ID,
-                    'name': MODEL_NAME,
+                'metric': {
+                    'id': 'jsd',
+                    'display_name': 'Jensen-Shannon Distance',
+                    'type': 'drift',
+                    'type_display_name': 'Data Drift',
                 },
-                'last_updated': '2023-11-28T08:04:10.918587+00:00',
             },
         ],
     }
@@ -177,7 +221,7 @@ LIST_API_RESPONSE_EMPTY = {
 @responses.activate
 def test_get_alert_rule_success() -> None:
     responses.get(
-        url=f'{URL}/v2/alert-configs/{ALERT_RULE_ID}',
+        url=f'{URL}/v3/alert-rules/{ALERT_RULE_ID}',
         json=API_RESPONSE_200,
     )
 
@@ -189,7 +233,7 @@ def test_get_alert_rule_success() -> None:
 @responses.activate
 def test_get_alert_rule_not_found() -> None:
     responses.get(
-        url=f'{URL}/v2/alert-configs/{ALERT_RULE_ID}',
+        url=f'{URL}/v3/alert-rules/{ALERT_RULE_ID}',
         json=API_RESPONSE_404,
         status=HTTPStatus.NOT_FOUND,
     )
@@ -201,7 +245,7 @@ def test_get_alert_rule_not_found() -> None:
 @responses.activate
 def test_alert_rule_list_success() -> None:
     responses.get(
-        url=f'{URL}/v2/alert-configs',
+        url=f'{URL}/v3/alert-rules',
         json=LIST_API_RESPONSE,
     )
     for rule in AlertRule.list(model_id=MODEL_ID):
@@ -212,7 +256,7 @@ def test_alert_rule_list_success() -> None:
 def test_alert_rule_list_empty() -> None:
 
     responses.get(
-        url=f'{URL}/v2/alert-configs',
+        url=f'{URL}/v3/alert-rules',
         json=LIST_API_RESPONSE_EMPTY,
     )
 
@@ -222,13 +266,13 @@ def test_alert_rule_list_empty() -> None:
 @responses.activate
 def test_delete_alert_rule() -> None:
     responses.get(
-        url=f'{URL}/v2/alert-configs/{ALERT_RULE_ID}',
+        url=f'{URL}/v3/alert-rules/{ALERT_RULE_ID}',
         json=API_RESPONSE_200,
     )
     rule = AlertRule.get(id_=ALERT_RULE_ID)
 
     responses.delete(
-        url=f'{URL}/v2/alert-configs/{ALERT_RULE_ID}',
+        url=f'{URL}/v3/alert-rules/{ALERT_RULE_ID}',
     )
 
     rule.delete()
@@ -237,13 +281,13 @@ def test_delete_alert_rule() -> None:
 @responses.activate
 def test_delete_alert_rule_not_found() -> None:
     responses.get(
-        url=f'{URL}/v2/alert-configs/{ALERT_RULE_ID}',
+        url=f'{URL}/v3/alert-rules/{ALERT_RULE_ID}',
         json=API_RESPONSE_200,
     )
     rule = AlertRule.get(id_=ALERT_RULE_ID)
 
     responses.delete(
-        url=f'{URL}/v2/alert-configs/{ALERT_RULE_ID}',
+        url=f'{URL}/v3/alert-rules/{ALERT_RULE_ID}',
         json=API_RESPONSE_404,
         status=HTTPStatus.NOT_FOUND,
     )
@@ -263,7 +307,7 @@ def test_add_alert_rule_success() -> None:
         json=BASELINE_API_RESPONSE_200,
     )
     responses.post(
-        url=f'{URL}/v2/alert-configs',
+        url=f'{URL}/v3/alert-rules',
         json=API_RESPONSE_200,
     )
     alert_rule = AlertRule(
@@ -288,7 +332,7 @@ def test_add_alert_rule_success() -> None:
 @responses.activate
 def test_enable_notifications(caplog) -> None:
     responses.get(
-        url=f'{URL}/v2/alert-configs/{ALERT_RULE_ID}',
+        url=f'{URL}/v3/alert-rules/{ALERT_RULE_ID}',
         json=API_RESPONSE_200,
     )
 
@@ -296,7 +340,7 @@ def test_enable_notifications(caplog) -> None:
 
     resp = responses.patch(
         url=f'{URL}/v2/alert-configs/{ALERT_RULE_ID}',
-        json=API_RESPONSE_200,
+        json=API_RESPONSE_200_V2,
     )
     set_logging(logging.INFO)
 
@@ -311,7 +355,7 @@ def test_enable_notifications(caplog) -> None:
 @responses.activate
 def test_disable_notifications(caplog) -> None:
     responses.get(
-        url=f'{URL}/v2/alert-configs/{ALERT_RULE_ID}',
+        url=f'{URL}/v3/alert-rules/{ALERT_RULE_ID}',
         json=API_RESPONSE_200,
     )
 
@@ -321,7 +365,7 @@ def test_disable_notifications(caplog) -> None:
 
     resp = responses.patch(
         url=f'{URL}/v2/alert-configs/{ALERT_RULE_ID}',
-        json=API_RESPONSE_200,
+        json=API_RESPONSE_200_V2,
     )
 
     set_logging(logging.INFO)
@@ -336,12 +380,12 @@ def test_disable_notifications(caplog) -> None:
 @responses.activate
 def test_set_notifications() -> None:
     responses.get(
-        url=f'{URL}/v2/alert-configs/{ALERT_RULE_ID}',
+        url=f'{URL}/v3/alert-rules/{ALERT_RULE_ID}',
         json=API_RESPONSE_200,
     )
 
     alert_rule = AlertRule.get(id_=ALERT_RULE_ID)
-    API_RESPONSE_200['data']['notifications'] = {
+    API_RESPONSE_200_V2['data']['notifications'] = {
         'webhooks': [
             {'uuid': 'e20bf4cc-d2cf-4540-baef-d96913b14f1b'},
             {'uuid': '6e796fda-0111-4a72-82cd-f0f219e903e1'},
@@ -358,7 +402,7 @@ def test_set_notifications() -> None:
     }
     responses.patch(
         url=f'{URL}/v2/alert-configs/{ALERT_RULE_ID}',
-        json=API_RESPONSE_200,
+        json=API_RESPONSE_200_V2,
     )
     notifications = alert_rule.set_notification_config(
         emails=['nikhil@fiddler.ai', 'admin@fiddler.ai'],
@@ -381,14 +425,14 @@ def test_set_notifications() -> None:
 @responses.activate
 def test_get_notifications() -> None:
     responses.get(
-        url=f'{URL}/v2/alert-configs/{ALERT_RULE_ID}',
+        url=f'{URL}/v3/alert-rules/{ALERT_RULE_ID}',
         json=API_RESPONSE_200,
     )
 
     alert_rule = AlertRule.get(id_=ALERT_RULE_ID)
     responses.get(
         url=f'{URL}/v2/alert-configs/{ALERT_RULE_ID}',
-        json=API_RESPONSE_200,
+        json=API_RESPONSE_200_V2,
     )
     notifications = alert_rule.get_notification_config()
     assert notifications == NotificationConfig(

{fiddler-client-3.1.0.dev3 → fiddler_client-3.1.0.dev4}/fiddler/tests/apis/test_model.py

@@ -672,3 +672,9 @@ def test_model_duplicate() -> None:
     model_copy = model.duplicate(version=new_version)
     assert model_copy.id is None
     assert model_copy.version == new_version
+
+    # Update new version
+    assert model.schema['CreditScore'].max == 850
+    model_copy.schema['CreditScore'].max = 900
+    assert model.schema['CreditScore'].max == 850
+    assert model_copy.schema['CreditScore'].max == 900

{fiddler-client-3.1.0.dev3 → fiddler_client-3.1.0.dev4}/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.1.0.dev3 → fiddler_client-3.1.0.dev4/fiddler_client.egg-info}/PKG-INFO

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

fiddler-client-3.1.0.dev3/fiddler/VERSION

@@ -1 +0,0 @@
-3.1.0.dev3