Merge pull request #83 from Scienfitz/main

fix: saving dataset via optional pandas / polars

docs/changelog.md

@@ -5,6 +5,11 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog],
 and this project adheres to [Semantic Versioning].
 
+## [2.1.11]
+
+## Fixed
+- `save_dataframe` incorrectly interacting with optional modules pandas or polars
+
 ## [2.1.10] - 2024-10-31
 
 ## Changed
@@ -306,6 +311,7 @@ and this project adheres to [Semantic Versioning].
 [Keep a Changelog]: https://keepachangelog.com/en/1.0.0/
 [Semantic Versioning]: https://semver.org/spec/v2.0.0.html
 
+[2.1.11]: https://github.com/emdgroup/foundry-dev-tools/compare/v2.1.10...v2.1.11
 [2.1.10]: https://github.com/emdgroup/foundry-dev-tools/compare/v2.1.9...v2.1.10
 [2.1.9]: https://github.com/emdgroup/foundry-dev-tools/compare/v2.1.8...v2.1.9
 [2.1.8]: https://github.com/emdgroup/foundry-dev-tools/compare/v2.1.7...v2.1.8

libs/foundry-dev-tools/src/foundry_dev_tools/resources/dataset.py

@@ -567,11 +567,13 @@ def save_dataframe(
             folder = str(round(time.time() * 1000)) if transaction_type == "APPEND" else "spark"
             parquet_compression = "snappy"
 
-            if not (pd.__fake__ or pl.__fake__) and isinstance(df, pd.DataFrame | pl.DataFrame):
+            use_pandas = not pd.__fake__ and isinstance(df, pd.DataFrame)
+            use_polars = not pl.__fake__ and isinstance(df, pl.DataFrame)
+            if use_pandas or use_polars:
                 buf = io.BytesIO()
                 schema_flavor = "spark"
 
-                if isinstance(df, pd.DataFrame):
+                if use_pandas:
                     # write pandas dataframe as parquet to buffer
                     df.to_parquet(
                         buf,