chore: add missing imports

Signed-off-by: Ajith Aravind <[email protected]>
unionai-oss · May 9, 2024 · c4b8e01 · c4b8e01
1 parent 361102e
commit c4b8e01
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/pandera/__init__.py b/pandera/__init__.py
@@ -62,12 +62,28 @@
     INT64,
     PANDAS_1_2_0_PLUS,
     PANDAS_1_3_0_PLUS,
+    PANDAS_2_0_0_PLUS,
     STRING,
     UINT8,
     UINT16,
     UINT32,
     UINT64,
     pandas_version,
+    ArrowBool,
+    ArrowDecimal128,
+    ArrowDictionary,
+    ArrowFloat32,
+    ArrowFloat64,
+    ArrowInt8,
+    ArrowInt16,
+    ArrowInt32,
+    ArrowInt64,
+    ArrowString,
+    ArrowTimestamp,
+    ArrowUInt8,
+    ArrowUInt16,
+    ArrowUInt32,
+    ArrowUInt64,
 )
 from pandera.schema_inference.pandas import infer_schema
 from pandera.version import __version__
@@ -136,12 +152,30 @@
     "INT16",
     "INT32",
     "INT64",
+    "PANDAS_1_2_0_PLUS",
     "PANDAS_1_3_0_PLUS",
+    "PANDAS_2_0_0_PLUS",
     "STRING",
     "UINT8",
     "UINT16",
     "UINT32",
     "UINT64",
+    # pandas_engine pyarrow
+    "ArrowBool",
+    "ArrowDecimal128",
+    "ArrowDictionary",
+    "ArrowFloat32",
+    "ArrowFloat64",
+    "ArrowInt8",
+    "ArrowInt16",
+    "ArrowInt32",
+    "ArrowInt64",
+    "ArrowString",
+    "ArrowTimestamp",
+    "ArrowUInt8",
+    "ArrowUInt16",
+    "ArrowUInt32",
+    "ArrowUInt64",
     # pandera.engines.pandas_engine
     "pandas_version",
     # checks