fiddler-client 3.2.0.dev3__tar.gz → 3.2.0.dev5__tar.gz

{fiddler_client-3.2.0.dev3/fiddler_client.egg-info → fiddler_client-3.2.0.dev5}/PKG-INFO

@@ -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.dev5/fiddler/VERSION

@@ -0,0 +1 @@
+3.2.0.dev5

{fiddler_client-3.2.0.dev3 → fiddler_client-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_client-3.2.0.dev3 → fiddler_client-3.2.0.dev5}/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_client-3.2.0.dev3 → fiddler_client-3.2.0.dev5}/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_client-3.2.0.dev3 → fiddler_client-3.2.0.dev5}/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 → fiddler_client-3.2.0.dev5/fiddler_client.egg-info}/PKG-INFO

@@ -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 → fiddler_client-3.2.0.dev5}/fiddler_client.egg-info/requires.txt

@@ -3,7 +3,7 @@ requests<3
 requests-toolbelt
 pandas>=1.2.5
 pydantic<2,>=1.10.15
-deprecated==1.2.13
+deprecated==1.2.14
 tqdm
 simplejson>=3.17.0
 pyarrow>=7.0.0

{fiddler_client-3.2.0.dev3 → fiddler_client-3.2.0.dev5}/setup.py

@@ -26,7 +26,7 @@ setuptools.setup(
         'requests-toolbelt',
         'pandas>=1.2.5',
         'pydantic>=1.10.15,<2',
-        'deprecated==1.2.13',
+        'deprecated==1.2.14',
         'tqdm',
         'simplejson>=3.17.0',
         'pyarrow>=7.0.0',

fiddler_client-3.2.0.dev3/fiddler/VERSION

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