create empty with pl

Signed-off-by: Alp Aribal <[email protected]>
unionai-oss · Aug 26, 2024 · 9981ee0 · 9981ee0
1 parent 13bb3ce
commit 9981ee0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pandera/api/polars/model.py b/pandera/api/polars/model.py
@@ -122,9 +122,9 @@ def to_json_schema(cls):
             FastAPI integration.
         """
         schema = cls.to_schema()
-        empty = pd.DataFrame(columns=schema.columns.keys()).astype(
-            {k: v.type for k, v in schema.dtypes.items()}
-        )
+        empty = pl.DataFrame(
+            schema={k: v.type for k, v in schema.dtypes.items()}
+        ).to_pandas()
         table_schema = pd.io.json.build_table_schema(empty)
 
         def _field_json_schema(field):