ygg 0.1.50__tar.gz → 0.1.51__tar.gz

{ygg-0.1.50 → ygg-0.1.51}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ygg
-Version: 0.1.50
+Version: 0.1.51
 Summary: Type-friendly utilities for moving data between Python objects, Arrow, Polars, Pandas, Spark, and Databricks
 Author: Yggdrasil contributors
 License:                                  Apache License

{ygg-0.1.50 → ygg-0.1.51}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ygg"
-version = "0.1.50"
+version = "0.1.51"
 description = "Type-friendly utilities for moving data between Python objects, Arrow, Polars, Pandas, Spark, and Databricks"
 readme = { file = "README.md", content-type = "text/markdown" }
 license = { file = "LICENSE" }

{ygg-0.1.50 → ygg-0.1.51}/src/ygg.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ygg
-Version: 0.1.50
+Version: 0.1.51
 Summary: Type-friendly utilities for moving data between Python objects, Arrow, Polars, Pandas, Spark, and Databricks
 Author: Yggdrasil contributors
 License:                                  Apache License

{ygg-0.1.50 → ygg-0.1.51}/src/yggdrasil/databricks/workspaces/io.py

@@ -666,10 +666,10 @@ class DatabricksIO(ABC, IO):
         self.seek(0)
 
         if isinstance(file_format, ParquetFileFormat):
-            pq.read_table(self, **kwargs)
+            return pq.read_table(self, **kwargs)
 
         elif isinstance(file_format, CsvFileFormat):
-            pcsv.read_csv(self, parse_options=file_format.parse_options)
+            return pcsv.read_csv(self, parse_options=file_format.parse_options)
 
         else:
             ValueError(f"Unsupported file format for Arrow table: {file_format}")
@@ -803,10 +803,10 @@ class DatabricksIO(ABC, IO):
         self.seek(0)
 
         if isinstance(file_format, ParquetFileFormat):
-            pandas.read_parquet(self, **kwargs)
+            return pandas.read_parquet(self, **kwargs)
 
         elif isinstance(file_format, CsvFileFormat):
-            pandas.read_csv(self, **kwargs)
+            return pandas.read_csv(self, **kwargs)
 
         else:
             raise ValueError(f"Unsupported file format for Pandas DataFrame: {file_format}")
@@ -865,10 +865,10 @@ class DatabricksIO(ABC, IO):
         self.seek(0)
 
         if isinstance(file_format, ParquetFileFormat):
-            polars.read_parquet(self, **kwargs)
+            return polars.read_parquet(self, **kwargs)
 
         elif isinstance(file_format, CsvFileFormat):
-            polars.read_csv(self, **kwargs)
+            return polars.read_csv(self, **kwargs)
 
         else:
             raise ValueError(f"Unsupported file format for Polars DataFrame: {file_format}")

{ygg-0.1.50 → ygg-0.1.51}/src/yggdrasil/databricks/workspaces/path.py

@@ -1121,9 +1121,9 @@ class DatabricksPath:
         """
         if self.is_file():
             with self.open("rb") as f:
-                return f.read_arrow_table(batch_size=batch_size, **kwargs)
-
-        if self.is_dir():
+                data = f.read_arrow_table(batch_size=batch_size, **kwargs)
+            return data
+        elif self.is_dir():
             tables: list[pa.Table] = []
             for child in self.ls(recursive=True):
                 if child.is_file():

ygg-0.1.51/src/yggdrasil/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.51"

ygg-0.1.50/src/yggdrasil/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.50"