fiddler-client 3.2.0.dev3__py3-none-any.whl → 3.2.0.dev5__py3-none-any.whl

fiddler/VERSION

@@ -1 +1 @@
-3.2.0.dev3
+3.2.0.dev5

fiddler/entities/alert_rule.py

@@ -36,6 +36,7 @@ class AlertRule(
         baseline_id: UUID | str | None = None,
         segment_id: UUID | str | None = None,
         compare_bin_delta: int | None = None,
+        evaluation_delay: int = 0,
     ) -> None:
         """Construct a alert rule instance."""
         self.name = name
@@ -46,6 +47,7 @@ class AlertRule(
         self.priority = priority
         self.compare_to = compare_to
         self.compare_bin_delta = compare_bin_delta
+        self.evaluation_delay = evaluation_delay
         self.warning_threshold = warning_threshold
         self.critical_threshold = critical_threshold
         self.condition = condition
@@ -95,6 +97,7 @@ class AlertRule(
             baseline_id=resp_obj.baseline.id if resp_obj.baseline else None,
             segment_id=resp_obj.segment.id if resp_obj.segment else None,
             compare_bin_delta=resp_obj.compare_bin_delta,
+            evaluation_delay=resp_obj.evaluation_delay,
         )
 
         # Add remaining fields
@@ -126,6 +129,7 @@ class AlertRule(
             'id',
             'created_at',
             'updated_at',
+            'evaluation_delay',
         ]
         for field in fields:
             setattr(self, field, getattr(resp_obj, field, None))
@@ -222,6 +226,7 @@ class AlertRule(
             'warning_threshold': self.warning_threshold,
             'feature_names': self.columns,
             'compare_bin_delta': self.compare_bin_delta,
+            'evaluation_delay': self.evaluation_delay,
         }
         if self.baseline_id:
             payload['baseline_id'] = self.baseline_id

fiddler/entities/events.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import os
 import tempfile
 from pathlib import Path
 from typing import Any, Callable
@@ -85,27 +86,40 @@ class EventPublisher(ConnectionMixin):
         dataset_name: str | None = None,
         update: bool = False,
     ) -> Job:
-        with tempfile.NamedTemporaryFile(suffix='.parquet') as temp_file:
-            try:
-                source.to_parquet(temp_file.name, index=False)
-                return self._publish_file(
-                    source=temp_file.name,
-                    environment=environment,
-                    dataset_name=dataset_name,
-                    update=update,
-                )
-            except Exception:
-                logger.warning(
-                    'Failed to convert input dataframe to parquet format. Retrying as a CSV file.'
-                )
-        with tempfile.NamedTemporaryFile(suffix='.csv') as temp_file:
+        temp_file = tempfile.NamedTemporaryFile(suffix='.parquet', delete=False)
+        try:
+            source.to_parquet(temp_file.name, index=False)
+            job = self._publish_file(
+                source=temp_file.name,
+                environment=environment,
+                dataset_name=dataset_name,
+                update=update,
+            )
+            temp_file.close()
+            return job
+
+        except Exception:
+            logger.warning(
+                'Failed to convert input dataframe to parquet format. Retrying as a CSV file.'
+            )
+
+        finally:
+            os.unlink(temp_file.name)
+
+        # Trying as csv file as source could not be converted to parquet
+        try:
+            temp_file = tempfile.NamedTemporaryFile(suffix='.csv', delete=False)
             source.to_csv(temp_file.name, index=False)
-            return self._publish_file(
+            job = self._publish_file(
                 source=temp_file.name,
                 environment=environment,
                 dataset_name=dataset_name,
                 update=update,
             )
+            temp_file.close()
+        finally:
+            os.unlink(temp_file.name)
+        return job
 
     def _publish_stream(
         self,

fiddler/schemas/alert_rule.py

@@ -26,6 +26,7 @@ class AlertRuleResp(BaseModel):
     compare_to: Union[str, CompareTo]
     condition: Union[str, AlertCondition]
     compare_bin_delta: Optional[int]
+    evaluation_delay: int
     bin_size: Union[str, BinSize]
     columns: Optional[List[str]] = Field(alias='feature_names')
     critical_threshold: float

fiddler/tests/apis/test_alert_rule.py

@@ -45,6 +45,7 @@ API_RESPONSE_200 = {
         'compare_to': 'raw_value',
         'condition': 'greater',
         'warning_threshold': 1.0,
+        'evaluation_delay': 10,
         'critical_threshold': 2.0,
         'updated_at': '2024-04-17T12:42:15.672909+00:00',
         'enable_notification': True,
@@ -183,6 +184,7 @@ LIST_API_RESPONSE = {
                 'updated_at': '2024-04-17T12:43:23.871997+00:00',
                 'enable_notification': True,
                 'compare_bin_delta': 0,
+                'evaluation_delay': 0,
                 'created_at': '2024-04-17T12:43:23.871997+00:00',
                 'created_by': {
                     'id': '287949c2-13f8-49fc-a559-55b2e6f36e9b',
@@ -321,12 +323,14 @@ def test_add_alert_rule_success() -> None:
         critical_threshold=0.5,
         warning_threshold=0.1,
         columns=['gender', 'creditscore'],
+        evaluation_delay=10,
     ).create()
 
     assert isinstance(alert_rule, AlertRule)
     assert alert_rule.id == UUID(ALERT_RULE_ID)
     assert alert_rule.model.id == UUID(MODEL_ID)
     assert alert_rule.project_id == UUID(PROJECT_ID)
+    assert alert_rule.evaluation_delay == 10
 
 
 @responses.activate

{fiddler_client-3.2.0.dev3.dist-info → fiddler_client-3.2.0.dev5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fiddler-client
-Version: 3.2.0.dev3
+Version: 3.2.0.dev5
 Summary: Python client for Fiddler Platform
 Home-page: https://fiddler.ai
 Author: Fiddler Labs
@@ -15,7 +15,7 @@ Requires-Dist: requests <3
 Requires-Dist: requests-toolbelt
 Requires-Dist: pandas >=1.2.5
 Requires-Dist: pydantic <2,>=1.10.15
-Requires-Dist: deprecated ==1.2.13
+Requires-Dist: deprecated ==1.2.14
 Requires-Dist: tqdm
 Requires-Dist: simplejson >=3.17.0
 Requires-Dist: pyarrow >=7.0.0

{fiddler_client-3.2.0.dev3.dist-info → fiddler_client-3.2.0.dev5.dist-info}/RECORD

@@ -1,4 +1,4 @@
-fiddler/VERSION,sha256=qP38gd73_0CmX2pYYwGWuMsLEV3njfXYC8AX_yLrGcI,11
+fiddler/VERSION,sha256=TRCU6c5adh6DHoLo2ldYt5MQ4P39J9008dME7ir402Y,11
 fiddler/__init__.py,sha256=pBLW7thJD1HkYhH5bJJbJFXSN8FNezos-rBU3HWhb4A,3794
 fiddler/configs.py,sha256=ZimSo0Gk7j1BFkjDHdRdycrGZ8oh-7G5TBXYiOh1HvU,217
 fiddler/connection.py,sha256=fLdSXpu5dwoK-h4Vtc9nbdKENOJ7jfEmrnDhVAtChck,5677
@@ -17,12 +17,12 @@ fiddler/constants/model_deployment.py,sha256=SfagLKVD7tK1QmYQFJLDrA4-6FaSeNF6tik
 fiddler/constants/xai.py,sha256=GR8VguB6FTfMJZ291FIbfDsY4S9BGOft3VxNW5jMf90,214
 fiddler/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 fiddler/entities/alert_record.py,sha256=RcDACWDNpntkiId-JpkLQvVeIbuaZAzn6AWl2zYIh9A,3564
-fiddler/entities/alert_rule.py,sha256=JQknBnSNnjIL4CeYvCBy5TVNN7LVE4-7BtBFpaQ9tJ0,11744
+fiddler/entities/alert_rule.py,sha256=ONBY11yZ54lTdmAqDYBZ6o4Ax7-7AFZS70UXY89SkCw,11971
 fiddler/entities/base.py,sha256=YCTLt1ta8UbD7u-5BW7v_ocA4S7LevEncI5XPS_SB60,2106
 fiddler/entities/baseline.py,sha256=k3D-K8WC5MkHDILyejRaH1v9jdCIymT96McDBOVQ9A0,7907
 fiddler/entities/custom_expression.py,sha256=SdS61JiI0gOxWwHAGxkHh1vqS5XsRtyeNSDs5E-Sg2U,6568
 fiddler/entities/dataset.py,sha256=kwD1pZMG6DQGIjejU7A2KUUBRgOi0o1qbbUX-PSlnGg,4399
-fiddler/entities/events.py,sha256=_orqFyg0m5tBREa21f36ZaX_4kKEtpA2Hsff0d__Uw0,5383
+fiddler/entities/events.py,sha256=9pzg7J1v03o2DGgHX_uPsCSujwk5iyPEDJPrrBNWJbo,5662
 fiddler/entities/file.py,sha256=dk8ISfjy0--F2eF23Ae8vGiEZmJW6e9LMqNmKfrY2mE,4710
 fiddler/entities/job.py,sha256=Feiqo6ns4bRcp2uaV-kuGYAKs90-aYXvzqqi1yLivYE,5220
 fiddler/entities/model.py,sha256=giqLWvnqXsqrekJ3sIeHf3vUUYO43PxrGh6I_KsGfH4,19271
@@ -45,7 +45,7 @@ fiddler/packtools/project_attributions_helpers.py,sha256=K-xpsKfwRS5GOMYwTmXY-J3
 fiddler/packtools/template_model.py,sha256=slQjl5eEjhDUtvWeIF_WvSUerX7VjW0822uo4Qkx-gg,10484
 fiddler/schemas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 fiddler/schemas/alert_record.py,sha256=Edfe264zUwqF856tMblN85CsEa_wn9oZCoLruvsNAbo,668
-fiddler/schemas/alert_rule.py,sha256=XbNZDBd5adLVvmyTydnG7lkdZ-ieI8maWACwBX-FQD0,1387
+fiddler/schemas/alert_rule.py,sha256=1MVVuylxT2t7Otn6phg4CIXJZzWBRDc7sfYiVw4ShGA,1413
 fiddler/schemas/base.py,sha256=upzAmneEz169h_w-UDH_tY0Abo7DeK68YOZ5qVwIMa8,189
 fiddler/schemas/baseline.py,sha256=tBQji1GLYt0ZYgEK6ZTmXxk5nUYTI6oxMmSjNXnRfWw,933
 fiddler/schemas/custom_expression.py,sha256=LGHht3LB0HGghEfy808VnaF3nIQVGZjhf__f1CcLRoU,953
@@ -78,7 +78,7 @@ fiddler/tests/test_utils.py,sha256=AF6y7UfboEkCa-nNw0aF30SdHC7yW2Hh78J4H_4s78k,3
 fiddler/tests/utils.py,sha256=FEDbASW4pYZktdEK2FLnIDyZCwBzpb60m6EzkvOkwSc,335
 fiddler/tests/apis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 fiddler/tests/apis/test_alert_record.py,sha256=_0EBRDn87TqTiO1KYEHKQQVtVZjj4lYjbt8Ix-RRTSg,3118
-fiddler/tests/apis/test_alert_rule.py,sha256=ax4-ravAzwbYbKay_Z8mksamTBaxlwZydFCD3EhLmfc,13196
+fiddler/tests/apis/test_alert_rule.py,sha256=5mou0sgwb7LEo_dirbz_6-cSoV7I6GD0r-hNzqvhPbk,13341
 fiddler/tests/apis/test_baseline.py,sha256=R_vvbZdSenpYMSydurENJrE_uNWE3sJR9sHJksfU_Jk,7494
 fiddler/tests/apis/test_custom_metric.py,sha256=QlNa80M9PxqaqzW-3nCtH5kKLRHYu2iqFDautBusbos,8208
 fiddler/tests/apis/test_dataset.py,sha256=1Xc9Ng91fLMc2C6dpKfse42fQ0anza3ZOKOD1TqCvTQ,4636
@@ -101,8 +101,8 @@ fiddler/utils/logger.py,sha256=FdJ3LkS9dbRjWsw5oJmNNsd7q0XRVEIvx5-TWys7L0k,669
 fiddler/utils/model_generator.py,sha256=TwQMQDpy-0gcq6HyL68s37N-sk_nGPq33mmppK6i7hI,2203
 fiddler/utils/validations.py,sha256=i8NtgrpCsitq6BPa5lygpKDh07oaOXu7PAlCIcMvqzY,408
 fiddler/utils/version.py,sha256=iC8Ry7UFl9EJW_xU1WbzO_l4yK6V7eQ6U5exB3xT864,531
-fiddler_client-3.2.0.dev3.dist-info/LICENSE.txt,sha256=w8-LUAb_VLBWSsCNmih0pAqLIicJfGu8OJXpDjkIg_o,559
-fiddler_client-3.2.0.dev3.dist-info/METADATA,sha256=s6-Lp2VoOIL8homkKGgcshE0xzMAJV8EuQykEvxrpYo,1602
-fiddler_client-3.2.0.dev3.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-fiddler_client-3.2.0.dev3.dist-info/top_level.txt,sha256=ndC6LqvKmrTTs8czRlBOYdYgSqZbHuMf0zHt_HWpzzk,8
-fiddler_client-3.2.0.dev3.dist-info/RECORD,,
+fiddler_client-3.2.0.dev5.dist-info/LICENSE.txt,sha256=w8-LUAb_VLBWSsCNmih0pAqLIicJfGu8OJXpDjkIg_o,559
+fiddler_client-3.2.0.dev5.dist-info/METADATA,sha256=Xp8FudH9TQkwOgw8QaBMqj5Cx8pddqDN-kiqjiofB_A,1602
+fiddler_client-3.2.0.dev5.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+fiddler_client-3.2.0.dev5.dist-info/top_level.txt,sha256=ndC6LqvKmrTTs8czRlBOYdYgSqZbHuMf0zHt_HWpzzk,8
+fiddler_client-3.2.0.dev5.dist-info/RECORD,,