Merge pull request #97 from JakobGM/thomasaarholt/docstrings

Fix docstring tests
JakobGM · Aug 18, 2024 · 477a242 · 477a242
2 parents ec9caf9 + 6dbf12b
commit 477a242
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/patito/polars.py b/src/patito/polars.py
@@ -817,8 +817,8 @@ def iter_models(
             ...     product_id: int = pt.Field(unique=True)
             ...     price: float
             ...
-            >>> df = pt.DataFrame({"product_id": [1, 2], "price": [10, 20]})
-            >>> df.set_model(Product)
+            >>> df = pt.DataFrame({"product_id": [1, 2], "price": [10., 20.]})
+            >>> df = df.set_model(Product)
             >>> for product in df.iter_models():
             ...     print(product)
             ...

diff --git a/src/patito/pydantic.py b/src/patito/pydantic.py
@@ -326,6 +326,7 @@ def from_row(
             >>> df = pl.DataFrame(
             ...     [["1", "product name", "1.22"]],
             ...     schema=["product_id", "name", "price"],
+            ...     orient="row",
             ... )
             >>> Product.from_row(df)
             Product(product_id=1, name='product name', price=1.22)
@@ -378,6 +379,7 @@ def _from_polars(
             >>> df = pl.DataFrame(
             ...     [["1", "product name", "1.22"]],
             ...     schema=["product_id", "name", "price"],
+            ...     orient="row",
             ... )
             >>> Product._from_polars(df)
             Product(product_id=1, name='product name', price=1.22)