add __all__ (#1797)

Signed-off-by: gabriel <[email protected]> Co-authored-by: gabriel <[email protected]>
unionai-oss · Aug 31, 2024 · ba6cd87 · ba6cd87
1 parent f277caa
commit ba6cd87
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/pandera/polars.py b/pandera/polars.py
@@ -1,6 +1,5 @@
 """A flexible and expressive polars validation library for Python."""
 
-# pylint: disable=unused-import
 from pandera import errors
 from pandera.api.checks import Check
 from pandera.api.dataframe.model_components import (
@@ -16,3 +15,20 @@
 from pandera.decorators import check_input, check_io, check_output, check_types
 
 register_polars_backends()
+
+
+__all__ = [
+    "check_input",
+    "check_io",
+    "check_output",
+    "check_types",
+    "check",
+    "Check",
+    "Column",
+    "dataframe_check",
+    "DataFrameModel",
+    "DataFrameSchema",
+    "errors",
+    "Field",
+    "PolarsData",
+]