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

{fiddler_client-3.2.0.dev4/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.dev4
+Version: 3.2.0.dev5
 Summary: Python client for Fiddler Platform
 Home-page: https://fiddler.ai
 Author: Fiddler Labs

fiddler_client-3.2.0.dev5/fiddler/VERSION

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

{fiddler_client-3.2.0.dev4 → 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.dev4 → 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.dev4
+Version: 3.2.0.dev5
 Summary: Python client for Fiddler Platform
 Home-page: https://fiddler.ai
 Author: Fiddler Labs

fiddler_client-3.2.0.dev4/fiddler/VERSION

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