Set no_implicit_reexport = true in pyproject.toml

py-polars/polars/__init__.py

@@ -11,7 +11,7 @@ def version() -> str:
     warnings.warn("polars binary missing!")
 
 import polars.testing as testing
-from polars.cfg import Config, toggle_string_cache  # We do not export in __all__
+from polars.cfg import Config
 from polars.convert import (
     from_arrow,
     from_dict,
@@ -122,6 +122,7 @@ def version() -> str:
     scan_ipc,
     scan_parquet,
 )
+from polars.string_cache import toggle_string_cache  # We do not export in __all__
 from polars.string_cache import StringCache
 from polars.utils import threadpool_size
 
@@ -133,6 +134,8 @@ def version() -> str:
     "ArrowError",
     "ComputeError",
     "NoDataError",
+    "DuplicateError",
+    "PanicException",
     "DataFrame",
     "Series",
     "LazyFrame",
@@ -154,6 +157,7 @@ def version() -> str:
     "Date",
     "Datetime",
     "Time",
+    "Duration",
     "Object",
     "Categorical",
     "Field",
@@ -186,6 +190,8 @@ def version() -> str:
     "date_range",
     "get_dummies",
     "repeat",
+    "element",
+    "cut",
     # polars.internal.lazy_functions
     "col",
     "count",
@@ -226,6 +232,8 @@ def version() -> str:
     "list",  # named to_list, see import above
     "select",
     "var",
+    "struct",
+    "duration",
     # polars.convert
     "from_dict",
     "from_dicts",

py-polars/polars/internals/__init__.py

@@ -12,7 +12,7 @@
 from .datatypes import IntoExpr
 from .expr import Expr, expr_to_lit_or_expr, selection_to_pyexpr_list, wrap_expr
 from .frame import DataFrame, wrap_df
-from .functions import concat, date_range  # DataFrame.describe() & DataFrame.upsample()
+from .functions import concat, date_range
 from .io import _is_local_file, _prepare_file_arg, read_ipc_schema, read_parquet_schema
 from .lazy_frame import LazyFrame, wrap_ldf
 from .lazy_functions import (
@@ -28,3 +28,34 @@
 )
 from .series import Series, wrap_s
 from .whenthen import when  # used in expr.clip()
+
+__all__ = [
+    "DataFrame",
+    "Expr",
+    "IntoExpr",
+    "LazyFrame",
+    "Series",
+    "all",
+    "arg_where",
+    "col",
+    "concat",
+    "concat_list",
+    "element",
+    "expr_to_lit_or_expr",
+    "format",
+    "lit",
+    "read_ipc_schema",
+    "read_parquet_schema",
+    "select",
+    "selection_to_pyexpr_list",
+    "when",
+    "wrap_df",
+    "wrap_expr",
+    "wrap_ldf",
+    "wrap_s",
+    "_is_local_file",
+    "_prepare_file_arg",
+    "_scan_ds",
+    "_scan_ipc_fsspec",
+    "_scan_parquet_fsspec",
+]

py-polars/polars/internals/frame.py

@@ -24,6 +24,7 @@
 from polars._html import NotebookFormatter
 from polars.datatypes import (
     Boolean,
+    ColumnsType,
     DataType,
     Int8,
     Int16,
@@ -38,7 +39,6 @@
     py_type_to_dtype,
 )
 from polars.internals.construction import (
-    ColumnsType,
     arrow_to_pydf,
     dict_to_pydf,
     numpy_to_pydf,

py-polars/pyproject.toml

@@ -38,7 +38,7 @@ strict_concatenate = true
 disallow_untyped_calls = true
 warn_redundant_casts = true
 # warn_return_any = true
-# no_implicit_reexport = true
+no_implicit_reexport = true
 # strict_equality = true
 # TODO: When all flags are enabled, replace by strict = true
 enable_error_code = [

py-polars/tests/io/test_avro.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
 
+import io
 import os
 
 import pytest
 
 import polars as pl
-from polars import io
 
 
 @pytest.fixture

py-polars/tests/test_lazy.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+from typing import Any, cast
+
 import numpy as np
 import pytest
 from _pytest.capture import CaptureFixture
@@ -879,7 +881,13 @@ def test_arithmetic() -> None:
 
 def test_ufunc() -> None:
     df = pl.DataFrame({"a": [1, 2]})
-    out = df.select(np.log(col("a")))  # type: ignore[call-overload]
+    # NOTE: unfortunately we must use cast instead of a type: ignore comment
+    #   1. CI job with Python 3.10, numpy==1.23.1 -> mypy complains about arg-type
+    #   2. so we try to resolve it with type: ignore[arg-type]
+    #   3. CI job with Python 3.7, numpy==1.21.6 -> mypy complains about
+    #       unused type: ignore comment
+    # for more information, see: https://github.com/python/mypy/issues/8823
+    out = df.select(np.log(cast(Any, col("a"))))
     assert out["a"][1] == 0.6931471805599453
 
 

py-polars/tests/test_series.py

@@ -2,7 +2,7 @@
 
 import math
 from datetime import date, datetime
-from typing import Any
+from typing import Any, cast
 from unittest.mock import patch
 
 import numpy as np
@@ -365,80 +365,80 @@ def test_ufunc() -> None:
     # test if output dtype is calculated correctly.
     s_float32 = pl.Series("a", [1.0, 2.0, 3.0, 4.0], dtype=pl.Float32)
     assert_series_equal(
-        np.multiply(s_float32, 4),  # type: ignore[arg-type]
+        cast(pl.Series, np.multiply(s_float32, 4)),
         pl.Series("a", [4.0, 8.0, 12.0, 16.0], dtype=pl.Float32),
     )
 
     s_float64 = pl.Series("a", [1.0, 2.0, 3.0, 4.0], dtype=pl.Float64)
     assert_series_equal(
-        np.multiply(s_float64, 4),  # type: ignore[arg-type]
+        cast(pl.Series, np.multiply(s_float64, 4)),
         pl.Series("a", [4.0, 8.0, 12.0, 16.0], dtype=pl.Float64),
     )
 
     s_uint8 = pl.Series("a", [1, 2, 3, 4], dtype=pl.UInt8)
     assert_series_equal(
-        np.power(s_uint8, 2),  # type: ignore[arg-type]
+        cast(pl.Series, np.power(s_uint8, 2)),
         pl.Series("a", [1, 4, 9, 16], dtype=pl.UInt8),
     )
     assert_series_equal(
-        np.power(s_uint8, 2.0),  # type: ignore[arg-type]
+        cast(pl.Series, np.power(s_uint8, 2.0)),
         pl.Series("a", [1.0, 4.0, 9.0, 16.0], dtype=pl.Float64),
     )
 
     s_int8 = pl.Series("a", [1, -2, 3, -4], dtype=pl.Int8)
     assert_series_equal(
-        np.power(s_int8, 2),  # type: ignore[arg-type]
+        cast(pl.Series, np.power(s_int8, 2)),
         pl.Series("a", [1, 4, 9, 16], dtype=pl.Int8),
     )
     assert_series_equal(
-        np.power(s_int8, 2.0),  # type: ignore[arg-type]
+        cast(pl.Series, np.power(s_int8, 2.0)),
         pl.Series("a", [1.0, 4.0, 9.0, 16.0], dtype=pl.Float64),
     )
 
     s_uint32 = pl.Series("a", [1, 2, 3, 4], dtype=pl.UInt32)
     assert_series_equal(
-        np.power(s_uint32, 2),  # type: ignore[arg-type]
+        cast(pl.Series, np.power(s_uint32, 2)),
         pl.Series("a", [1, 4, 9, 16], dtype=pl.UInt32),
     )
     assert_series_equal(
-        np.power(s_uint32, 2.0),  # type: ignore[arg-type]
+        cast(pl.Series, np.power(s_uint32, 2.0)),
         pl.Series("a", [1.0, 4.0, 9.0, 16.0], dtype=pl.Float64),
     )
 
     s_int32 = pl.Series("a", [1, -2, 3, -4], dtype=pl.Int32)
     assert_series_equal(
-        np.power(s_int32, 2),  # type: ignore[arg-type]
+        cast(pl.Series, np.power(s_int32, 2)),
         pl.Series("a", [1, 4, 9, 16], dtype=pl.Int32),
     )
     assert_series_equal(
-        np.power(s_int32, 2.0),  # type: ignore[arg-type]
+        cast(pl.Series, np.power(s_int32, 2.0)),
         pl.Series("a", [1.0, 4.0, 9.0, 16.0], dtype=pl.Float64),
     )
 
     s_uint64 = pl.Series("a", [1, 2, 3, 4], dtype=pl.UInt64)
     assert_series_equal(
-        np.power(s_uint64, 2),  # type: ignore[arg-type]
+        cast(pl.Series, np.power(s_uint64, 2)),
         pl.Series("a", [1, 4, 9, 16], dtype=pl.UInt64),
     )
     assert_series_equal(
-        np.power(s_uint64, 2.0),  # type: ignore[arg-type]
+        cast(pl.Series, np.power(s_uint64, 2.0)),
         pl.Series("a", [1.0, 4.0, 9.0, 16.0], dtype=pl.Float64),
     )
 
     s_int64 = pl.Series("a", [1, -2, 3, -4], dtype=pl.Int64)
     assert_series_equal(
-        np.power(s_int64, 2),  # type: ignore[arg-type]
+        cast(pl.Series, np.power(s_int64, 2)),
         pl.Series("a", [1, 4, 9, 16], dtype=pl.Int64),
     )
     assert_series_equal(
-        np.power(s_int64, 2.0),  # type: ignore[arg-type]
+        cast(pl.Series, np.power(s_int64, 2.0)),
         pl.Series("a", [1.0, 4.0, 9.0, 16.0], dtype=pl.Float64),
     )
 
     # test if null bitmask is preserved
     a1 = pl.Series("a", [1.0, None, 3.0])
-    b1 = np.exp(a1)
-    assert b1.null_count() == 1  # type: ignore[attr-defined]
+    b1 = cast(pl.Series, np.exp(a1))
+    assert b1.null_count() == 1
 
     # test if it works with chunked series.
     a2 = pl.Series("a", [1.0, None, 3.0])
@@ -447,7 +447,7 @@ def test_ufunc() -> None:
     assert a2.n_chunks() == 2
     c2 = np.multiply(a2, 3)
     assert_series_equal(
-        c2,  # type: ignore[arg-type]
+        cast(pl.Series, c2),
         pl.Series("a", [3.0, None, 9.0, 12.0, 15.0, None]),
     )
 
@@ -1150,15 +1150,12 @@ def test_abs() -> None:
     # ints
     s = pl.Series([1, -2, 3, -4])
     assert_series_equal(s.abs(), pl.Series([1, 2, 3, 4]))
-    assert_series_equal(np.abs(s), pl.Series([1, 2, 3, 4]))  # type: ignore[arg-type]
+    assert_series_equal(cast(pl.Series, np.abs(s)), pl.Series([1, 2, 3, 4]))
 
     # floats
     s = pl.Series([1.0, -2.0, 3, -4.0])
     assert_series_equal(s.abs(), pl.Series([1.0, 2.0, 3.0, 4.0]))
-    assert_series_equal(
-        np.abs(s),  # type: ignore[arg-type]
-        pl.Series([1.0, 2.0, 3.0, 4.0]),
-    )
+    assert_series_equal(cast(pl.Series, np.abs(s)), pl.Series([1.0, 2.0, 3.0, 4.0]))
     assert_series_equal(
         pl.select(pl.lit(s).abs()).to_series(), pl.Series([1.0, 2.0, 3.0, 4.0])
     )