From 1100fb24ba090b39c90e05239762210dbd7dc9c5 Mon Sep 17 00:00:00 2001
From: gabriel <gabriel.g.robin@airbus.com>
Date: Fri, 30 Aug 2024 22:25:28 +0200
Subject: [PATCH] add __all__

Signed-off-by: gabriel <gabriel.g.robin@airbus.com>
---
 pandera/polars.py | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/pandera/polars.py b/pandera/polars.py
index b9ea0b490..431103a9e 100644
--- 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",
+]