fiddler-client 2.4.0.dev2__py3-none-any.whl → 2.5.0.dev2__py3-none-any.whl

tests/fiddler/test_alert.py

@@ -10,6 +10,7 @@ from fiddler.exceptions import NotFound
 from fiddler.schema.alert import (
     AlertCondition,
     AlertRule,
+    AlertRuleWithNotifications,
     AlertType,
     BinSize,
     ComparePeriod,
@@ -1185,6 +1186,8 @@ class TestAlert(BaseTestCase):
                         'severity': None,
                         'failure_reason': 'NA',
                         'message': 'In project bank_churn and model  bank_churn, during the time period of one hour  starting |2022-07-13 11:00:00 (UTC time)|, the prediction drift score was 0.789,   greater than 0.1.',
+                        'alert_record_main_version': 1,
+                        'alert_record_sub_version': 1,
                     },
                     {
                         'id': 2,
@@ -1199,6 +1202,8 @@ class TestAlert(BaseTestCase):
                         'severity': None,
                         'failure_reason': 'NA',
                         'message': 'In project bank_churn and model  bank_churn, during the time period of one hour  starting |2022-07-13 14:00:00 (UTC time)|, the prediction drift score was 0.789,   greater than 0.1.',
+                        'alert_record_main_version': 1,
+                        'alert_record_sub_version': 1,
                     },
                     {
                         'id': 3,
@@ -1214,6 +1219,8 @@ class TestAlert(BaseTestCase):
                         'failure_reason': 'NA',
                         'message': 'In project bank_churn and model  bank_churn, during the time period of one hour  starting |2022-07-13 15:00:00 (UTC time)|, the prediction drift score was 0.789,   greater than 0.1.',
                         'feature_name': 'feature_name',
+                        'alert_record_main_version': 1,
+                        'alert_record_sub_version': 1,
                     },
                 ],
             }
@@ -1397,7 +1404,6 @@ class TestAlert(BaseTestCase):
         self.assertEqual(alert_rule_parsed_frequency.column, 'geography')
         self.assertEqual(alert_rule_parsed_frequency.category, 'Hawaii')
 
-
     def test_triggered_alert_record_serializer(self) -> None:
         tar_with_value_as_float = {
             'id': 1,
@@ -1414,6 +1420,8 @@ class TestAlert(BaseTestCase):
             'message': 'some message',
             'multi_col_values': {'a': 1.0, 'b': 2.0},
             'feature_name': 'age',
+            'alert_record_main_version': 1,
+            'alert_record_sub_version': 1,
         }
         tar: TriggeredAlerts = TriggeredAlerts.parse_obj(tar_with_value_as_float)
         self.assertEqual(tar.alert_value, 1.0)
@@ -1432,17 +1440,99 @@ class TestAlert(BaseTestCase):
             'severity': 'HIGH',
             'failure_reason': 'NA',
             'message': 'some message',
+            'alert_record_main_version': 1,
+            'alert_record_sub_version': 1,
         }
         tar_dict: TriggeredAlerts = TriggeredAlerts.parse_obj(tar_with_value_as_dict)
         self.assertEqual(tar_dict.alert_value, {'a': 1.0, 'b': 2.0})
 
     def test_enum_values(self) -> None:
-        metric_values = [item.value for item in Metric] 
-        for metric in ['null_violation_count', 'range_violation_count', 'type_violation_count', 
-            'expected_calibration_error', 'calibrated_threshold',
-            'geometric_mean', 'ndcg_mean', 'traffic', 'frequency',]:
+        metric_values = [item.value for item in Metric]
+        for metric in [
+            'null_violation_count',
+            'range_violation_count',
+            'type_violation_count',
+            'expected_calibration_error',
+            'calibrated_threshold',
+            'geometric_mean',
+            'ndcg_mean',
+            'traffic',
+            'frequency',
+        ]:
             self.assertIn(metric, metric_values)
 
+    def test_update_alert_notification_status(self) -> None:
+        response = {
+            'data': {
+                'created_by': 'admin@fiddler.ai',
+                'priority': 'LOW',
+                'alert_type': 'drift',
+                'compare_period': 86400000,
+                'time_bucket': 3600000,
+                'metric': 'jsd',
+                'model_name': self._model_id,
+                'organization_name': self._org,
+                'created_at': '2022-09-15T18:10:38.286686+00:00',
+                'critical_threshold': 0.2,
+                'feature_names': ['age'],
+                'is_active': True,
+                'enable_notification': True,
+                'uuid': '9f8180d3-3fa0-40c4-8656-b9b1d2de1b69',
+                'compare_to': 'time_period',
+                'warning_threshold': 0.7,
+                'condition': 'lesser',
+                'project_name': self._project_id,
+                'id': 86,
+                'name': 'new_name',
+                'metric_display_name': 'Jenson-Shannon Distance',
+                'alert_type_display_name': 'Data Drift',
+                'notifications': {
+                    'emails': {'email': 'nikhil@fiddler.ai, admin@fiddler.ai'},
+                    'pagerduty': {
+                        'service': 'pd_test_1, pd_test_2',
+                        'severity': 'critical',
+                    },
+                    'webhooks': [{'uuid': '5e4eceae-0387-44fe-b726-7516c33e33fe'}],
+                },
+            },
+            'api_version': '2.0',
+            'kind': 'NORMAL',
+        }
+
+        url = self._get_url(response['data']['uuid'])
+
+        self.requests_mock.patch(
+            url,
+            json=response,
+        )
+
+        self.requests_mock.get(
+            url,
+            json=response,
+        )
+
+        # columns parameter is supported from 23.3.0
+        server_info = ServerInfo(**{'feature_flags': {}, 'server_version': '23.3.0'})
+        with mock.patch.object(
+            self.client, 'server_info', server_info
+        ) as mock_server_info:
+            alert_rules: list[
+                AlertRuleWithNotifications
+            ] = self.client.update_alert_notification_status(
+                alert_config_ids=[response['data']['uuid']], notification_status=True
+            )
+        alert_rule = alert_rules[0]
+        self.assertEqual(
+            alert_rule.alert_rule_uuid, '9f8180d3-3fa0-40c4-8656-b9b1d2de1b69'
+        )
+        self.assertEqual(alert_rule.critical_threshold, 0.2)
+        self.assertEqual(alert_rule.name, 'new_name')
+        self.assertEqual(alert_rule.columns, ['age'])
+        self.assertEqual(alert_rule.compare_period, ComparePeriod.ONE_DAY)
+        self.assertIsNone(alert_rule.baseline_name)
+
+        self.assertEqual(len(self.requests_mock.calls), 2)
+
 
 if __name__ == '__main__':
     unittest.main()

fiddler3/entities/helpers.py

@@ -1,21 +0,0 @@
-from http import HTTPStatus
-
-from fiddler3.exceptions import NotFound
-from fiddler3.schemas.response import ErrorData
-
-
-def raise_not_found(message: str) -> None:
-    """Raise NotFound if the resource is not found while fetching with names"""
-    raise NotFound(
-        error=ErrorData(
-            code=HTTPStatus.NOT_FOUND,
-            message=message,
-            errors=[
-                {
-                    'reason': 'ObjectNotFound',  # type: ignore
-                    'message': message,
-                    'help': '',
-                }
-            ],
-        ),
-    )

fiddler3/schemas/model_artifact.py

@@ -1,6 +0,0 @@
-from fiddler3.schemas.base import BaseModel
-
-
-class ModelArtifactDeployMultiPartUploadResp(BaseModel):
-    part_number: int
-    etag: str