patito 0.8.3__py3-none-any.whl → 0.8.5__py3-none-any.whl

patito/_pydantic/repr.py

@@ -90,7 +90,7 @@ class Representation:
         return self.__repr_str__(" ")
 
     def __repr__(self) -> str:
-        return f'{self.__repr_name__()}({self.__repr_str__(", ")})'
+        return f"{self.__repr_name__()}({self.__repr_str__(', ')})"
 
     def __rich_repr__(self) -> "RichReprResult":
         """Get fields for Rich library."""

patito/exceptions.py

@@ -79,7 +79,7 @@ class DataFrameValidationError(Representation, ValueError):
         errors = self.errors()
         no_errors = len(errors)
         return (
-            f'{no_errors} validation error{"" if no_errors == 1 else "s"} for {self.model.__name__}\n'
+            f"{no_errors} validation error{'' if no_errors == 1 else 's'} for {self.model.__name__}\n"
             f"{display_errors(errors)}"
         )
 
@@ -90,7 +90,7 @@ class DataFrameValidationError(Representation, ValueError):
 
 def display_errors(errors: list["ErrorDict"]) -> str:
     return "\n".join(
-        f'{_display_error_loc(e)}\n  {e["msg"]} ({_display_error_type_and_ctx(e)})'
+        f"{_display_error_loc(e)}\n  {e['msg']} ({_display_error_type_and_ctx(e)})"
         for e in errors
     )
 

patito/polars.py

@@ -213,7 +213,7 @@ class LazyFrame(pl.LazyFrame, Generic[ModelType]):
             lf = lf.with_columns(derived_from.cast(dtype).alias(column_name))
         else:
             raise TypeError(
-                "Can not derive dataframe column from type " f"{type(derived_from)}."
+                f"Can not derive dataframe column from type {type(derived_from)}."
             )
         derived_columns.append(column_name)
         return lf, derived_columns
@@ -337,7 +337,7 @@ class LazyFrame(pl.LazyFrame, Generic[ModelType]):
     def from_existing(cls: type[LDF], lf: pl.LazyFrame) -> LDF:
         """Construct a patito.DataFrame object from an existing polars.DataFrame object."""
         if getattr(cls, "model", False):
-            return cls.model.LazyFrame._from_pyldf(super().lazy()._ldf)  # type: ignore
+            return cls.model.LazyFrame._from_pyldf(super().lazy(lf)._ldf)  # type: ignore
 
         return LazyFrame._from_pyldf(lf._ldf)  # type: ignore
 
@@ -443,7 +443,7 @@ class DataFrame(pl.DataFrame, Generic[ModelType]):
             >>> casted_classes.validate()
 
         """
-        return model.DataFrame(self._df)
+        return model.DataFrame._from_pydf(self._df)
 
     def unalias(self: DF) -> DF:
         """Un-aliases column names using information from pydantic validation_alias.
@@ -537,9 +537,8 @@ class DataFrame(pl.DataFrame, Generic[ModelType]):
 
         """
         if columns is not None:
-            # I get a single null row if I try to use super() here, so go via
-            # pl.DataFrame instead.
-            return self._from_pydf(pl.DataFrame(self._df).drop(columns)._df)
+            # Use super() to call polars DataFrame drop method directly
+            return self._from_pydf(super().drop(columns)._df)
         else:
             return self.drop(list(set(self.columns) - set(self.model.columns)))
 

patito/validators.py

@@ -68,9 +68,9 @@ def _transform_df(dataframe: pl.DataFrame, schema: type[Model]) -> pl.DataFrame:
     if alias_gen := schema.model_config.get("alias_generator"):
         if isinstance(alias_gen, AliasGenerator):
             alias_func = alias_gen.validation_alias or alias_gen.alias
-            assert (
-                alias_func is not None
-            ), "An AliasGenerator must contain a transforming function"
+            assert alias_func is not None, (
+                "An AliasGenerator must contain a transforming function"
+            )
         else:  # alias_gen is a function
             alias_func = alias_gen
 
@@ -129,7 +129,7 @@ def _find_errors(  # noqa: C901
                 )
             )
 
-    if not allow_superfluous_columns:
+    if not (allow_superfluous_columns or schema.model_config.get("extra") == "allow"):
         # Check if any additional columns are included
         for superfluous_column in set(column_subset) - set(schema.columns):
             errors.append(

{patito-0.8.3.dist-info → patito-0.8.5.dist-info}/METADATA

@@ -1,35 +1,22 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: patito
-Version: 0.8.3
+Version: 0.8.5
 Summary: A dataframe modelling library built on top of polars and pydantic.
-Home-page: https://github.com/JakobGM/patito
-License: MIT
-Keywords: validation,dataframe
-Author: Jakob Gerhard Martinussen
-Author-email: jakobgm@gmail.com
+Project-URL: Homepage, https://github.com/JakobGM/patito
+Project-URL: Repository, https://github.com/JakobGM/patito
+Project-URL: Documentation, https://patito.readthedocs.io
+Author-email: Jakob Gerhard Martinussen <jakobgm@gmail.com>, Thomas Aarholt <thomasaarholt@gmail.com>
+License-Expression: MIT
+License-File: LICENSE
+Keywords: dataframe,validation
 Requires-Python: >=3.9
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: polars>=1.32.0
+Requires-Dist: pydantic>=2.7.0
+Requires-Dist: typing-extensions
 Provides-Extra: caching
-Provides-Extra: docs
+Requires-Dist: pyarrow>=5.0.0; extra == 'caching'
 Provides-Extra: pandas
-Requires-Dist: Sphinx (<7) ; extra == "docs"
-Requires-Dist: pandas ; extra == "pandas"
-Requires-Dist: polars (>=1.10.0)
-Requires-Dist: pyarrow (>=5.0.0) ; extra == "caching"
-Requires-Dist: pydantic (>=2.7.0)
-Requires-Dist: sphinx-autobuild ; extra == "docs"
-Requires-Dist: sphinx-autodoc-typehints ; extra == "docs"
-Requires-Dist: sphinx-rtd-theme ; extra == "docs"
-Requires-Dist: sphinx-toolbox ; extra == "docs"
-Requires-Dist: sphinxcontrib-mermaid ; extra == "docs"
-Requires-Dist: typing-extensions
-Project-URL: Documentation, https://patito.readthedocs.io
-Project-URL: Repository, https://github.com/JakobGM/patito
+Requires-Dist: pandas; extra == 'pandas'
 Description-Content-Type: text/markdown
 
 # <center><img height="30px" src="https://em-content.zobj.net/thumbs/120/samsung/78/duck_1f986.png"> Patito<center>
@@ -307,4 +294,3 @@ milk = products.get(pl.col("product_id") == 1)
 print(milk.url)
 # https://example.com/no/products/1-skimmed-milk
 ```
-

{patito-0.8.3.dist-info → patito-0.8.5.dist-info}/RECORD

@@ -1,17 +1,17 @@
 patito/__init__.py,sha256=4qD13kfoa85_kyTCChm3xQcKKzIy3G8AZQp8T_bjcmo,844
 patito/_docs.py,sha256=9mfttyylWpqaOZv8xfDMEwCHHaY7GQwfyI7CDg7tWe8,162
+patito/exceptions.py,sha256=rv0AUVqxxnP00wmjgaqOPz0Ht6er-SjdeunHAul9brs,6034
+patito/polars.py,sha256=1AYcsJRItAnMt5TN5NI-w4plMVauGhf_HFl7rhqgb7E,38743
+patito/pydantic.py,sha256=yrpbjjqu8fRJkoPWt_uCAfs2yNDDBB1OviZwxnnPKpI,49456
+patito/validators.py,sha256=Ne6mj8js3RDAcgw6p558wdic3VA7Kbu6AgDIq0X9hzg,18603
 patito/_pydantic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 patito/_pydantic/column_info.py,sha256=u6sZ2212VppgBmcISWzGNY6ZfBiNFJtlwj1lDXDFL4c,5366
+patito/_pydantic/repr.py,sha256=qMQwkEzfr9t4aIo1O2IxHC5r2QreuxYNQKT5eddkDQs,4182
+patito/_pydantic/schema.py,sha256=BI2qAhNM29NxS366K9eRi8thgE2P3t8GFt1HzwlWxos,3603
 patito/_pydantic/dtypes/__init__.py,sha256=2vTvL4N4yMN0cbv2CSoa1OFwCswx6FhuQdsYhMaz_dU,578
 patito/_pydantic/dtypes/dtypes.py,sha256=KETa3U01sQEub6sCckCHfNz781LOlvziRwxADkyUQyA,9842
 patito/_pydantic/dtypes/utils.py,sha256=wuJv-cYDftaN1OLCGM4sPlEMsIj7rZLlaHKezBa474U,6659
-patito/_pydantic/repr.py,sha256=P7ojqTeNM4htzZgw2qMO6XzFqkiIXLwP-WUIEqNt7I0,4182
-patito/_pydantic/schema.py,sha256=BI2qAhNM29NxS366K9eRi8thgE2P3t8GFt1HzwlWxos,3603
-patito/exceptions.py,sha256=XK6UF_UojeOR45TJnZqS19SHZAUIvu-nswqf1tnFJ08,6034
-patito/polars.py,sha256=Zqk6m26EjRQwp6Ug0UALyYDNT9cbz-GyqZ8_uLWuFwU,38790
-patito/pydantic.py,sha256=yrpbjjqu8fRJkoPWt_uCAfs2yNDDBB1OviZwxnnPKpI,49456
-patito/validators.py,sha256=g6zAzV-C50ffkNkHV1FzGzJKeaBzs09DK2liXsnHphI,18554
-patito-0.8.3.dist-info/LICENSE,sha256=3bc4YyuF0e5nd59E3CsR8QM1Ua7pqKfC9DD1LVBVMs4,1139
-patito-0.8.3.dist-info/METADATA,sha256=G_MOL6X9cvOZrvjxvqS4-Eo-xYjO0jG5kw61c4cgMOs,13999
-patito-0.8.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-patito-0.8.3.dist-info/RECORD,,
+patito-0.8.5.dist-info/METADATA,sha256=h65iK0hXtx89a7Srl2qGYrNJ4c1g1Bg-OHW0KS7qZg4,13439
+patito-0.8.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+patito-0.8.5.dist-info/licenses/LICENSE,sha256=3bc4YyuF0e5nd59E3CsR8QM1Ua7pqKfC9DD1LVBVMs4,1139
+patito-0.8.5.dist-info/RECORD,,

{patito-0.8.3.dist-info → patito-0.8.5.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.0
+Generator: hatchling 1.27.0
 Root-Is-Purelib: true
 Tag: py3-none-any