ygg 0.1.50__py3-none-any.whl → 0.1.51__py3-none-any.whl

{ygg-0.1.50.dist-info → ygg-0.1.51.dist-info}/METADATA

@@ -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.dist-info → ygg-0.1.51.dist-info}/RECORD

@@ -1,6 +1,6 @@
-ygg-0.1.50.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+ygg-0.1.51.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
 yggdrasil/__init__.py,sha256=4-ghPak2S6zfMqmnlxW2GCgPb5s79znpKa2hGEGXcE4,24
-yggdrasil/version.py,sha256=pMWaMbj0sqJPaN27zeKuthOtJ3nuofEVeTxWuJmKhTw,22
+yggdrasil/version.py,sha256=Vba463sBalMddSSVNE_881HL99Fg7msGZpAiG0JX6bg,22
 yggdrasil/databricks/__init__.py,sha256=skctY2c8W-hI81upx9F_PWRe5ishL3hrdiTuizgDjdw,152
 yggdrasil/databricks/compute/__init__.py,sha256=NvdzmaJSNYY1uJthv1hHdBuNu3bD_-Z65DWnaJt9yXg,289
 yggdrasil/databricks/compute/cluster.py,sha256=YomLfvB0oxbgl6WDgBRxI1UXsxwlEbR6gq3FUbPHscY,44199
@@ -16,8 +16,8 @@ yggdrasil/databricks/sql/types.py,sha256=5G-BM9_eOsRKEMzeDTWUsWW5g4Idvs-czVCpOCr
 yggdrasil/databricks/sql/warehouse.py,sha256=1J0dyQLJb-OS1_1xU1eAVZ4CoL2-FhFeowKSvU3RzFc,9773
 yggdrasil/databricks/workspaces/__init__.py,sha256=dv2zotoFVhNFlTCdRq6gwf5bEzeZkOZszoNZMs0k59g,114
 yggdrasil/databricks/workspaces/filesytem.py,sha256=Z8JXU7_XUEbw9fpTQT1avRQKi-IAP2KemXBMPkUoY4w,9805
-yggdrasil/databricks/workspaces/io.py,sha256=D-B31roMGEJesAtUWl-O30lZJfgo-oFdK6KExzFc13I,33260
-yggdrasil/databricks/workspaces/path.py,sha256=BAzaxEL2mWJ_6EnETnQdsPj06zkrbTO2f3reruR439k,49265
+yggdrasil/databricks/workspaces/io.py,sha256=PhXMVrK8ngDl6kKjnh8_jlZ2GsKtU2nLSi1nFgV4Sks,33302
+yggdrasil/databricks/workspaces/path.py,sha256=HA73r0qedm8IiE_FPrDzRLc5BBkU9_a1qF2JXdWXMQk,49290
 yggdrasil/databricks/workspaces/path_kind.py,sha256=Xc319NysH8_6E9C0Q8nCxDHYG07_SnzyUVKHe0dNdDQ,305
 yggdrasil/databricks/workspaces/workspace.py,sha256=zBlQdYNT_xKwUCYo3O4Q4g-8pfMvff3I26efyCfY_TY,24961
 yggdrasil/dataclasses/__init__.py,sha256=_RkhfF3KC1eSORby1dzvBXQ0-UGG3u6wyUQWX2jq1Pc,108
@@ -55,8 +55,8 @@ yggdrasil/types/cast/registry.py,sha256=_zdFGmUBB7P-e_LIcJlOxMcxAkXoA-UXB6HqLMgT
 yggdrasil/types/cast/spark_cast.py,sha256=_KAsl1DqmKMSfWxqhVE7gosjYdgiL1C5bDQv6eP3HtA,24926
 yggdrasil/types/cast/spark_pandas_cast.py,sha256=BuTiWrdCANZCdD_p2MAytqm74eq-rdRXd-LGojBRrfU,5023
 yggdrasil/types/cast/spark_polars_cast.py,sha256=btmZNHXn2NSt3fUuB4xg7coaE0RezIBdZD92H8NK0Jw,9073
-ygg-0.1.50.dist-info/METADATA,sha256=ygOCZJjNIbuKuD-qKLnttguy71qIBxR0KnHDJE_XPSU,18528
-ygg-0.1.50.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ygg-0.1.50.dist-info/entry_points.txt,sha256=6q-vpWG3kvw2dhctQ0LALdatoeefkN855Ev02I1dKGY,70
-ygg-0.1.50.dist-info/top_level.txt,sha256=iBe9Kk4VIVbLpgv_p8OZUIfxgj4dgJ5wBg6vO3rigso,10
-ygg-0.1.50.dist-info/RECORD,,
+ygg-0.1.51.dist-info/METADATA,sha256=JprFwC_aHRV7jMw6YBV4-uAZcTZrEFVu7eE6_2dulG4,18528
+ygg-0.1.51.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ygg-0.1.51.dist-info/entry_points.txt,sha256=6q-vpWG3kvw2dhctQ0LALdatoeefkN855Ev02I1dKGY,70
+ygg-0.1.51.dist-info/top_level.txt,sha256=iBe9Kk4VIVbLpgv_p8OZUIfxgj4dgJ5wBg6vO3rigso,10
+ygg-0.1.51.dist-info/RECORD,,

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}")

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():

yggdrasil/version.py

@@ -1 +1 @@
-__version__ = "0.1.50"
+__version__ = "0.1.51"