explicitly validate row dict

unionai-oss · Jun 19, 2024 · ce0553f · ce0553f
1 parent efbaa91
commit ce0553f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pandera/engines/pandas_engine.py b/pandera/engines/pandas_engine.py
@@ -1318,9 +1318,9 @@ def _coerce_row(row):
             try:
                 # pylint: disable=not-callable
                 if PYDANTIC_V2:
-                    row = self.type(**row).model_dump()
+                    row = self.type.model_validate(row).model_dump()
                 else:
-                    row = self.type(**row).dict()
+                    row = self.type.parse_obj(row).dict()
                 row["failure_cases"] = np.nan
             except ValidationError as exc:
                 row["failure_cases"] = {