Clean internals imports (#1847)

pola-rs · Nov 21, 2021 · 6aca48d · 6aca48d
1 parent 3a2fad9
commit 6aca48d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/py-polars/polars/internals/__init__.py b/py-polars/polars/internals/__init__.py
@@ -5,8 +5,8 @@
 """
 from .expr import Expr, _selection_to_pyexpr_list, expr_to_lit_or_expr, wrap_expr
 from .frame import DataFrame, wrap_df
-from .functions import arg_where, concat, date_range, get_dummies, repeat
+from .functions import concat, date_range  # DataFrame.describe() & DataFrame.upsample()
 from .lazy_frame import LazyFrame, wrap_ldf
-from .lazy_functions import arange, argsort_by, col, concat_list, lit
+from .lazy_functions import argsort_by, col, concat_list, lit
 from .series import Series, wrap_s
 from .whenthen import when  # used in expr.clip()
diff --git a/py-polars/polars/internals/lazy_functions.py b/py-polars/polars/internals/lazy_functions.py
@@ -838,7 +838,7 @@ def arange(
 
     if eager:
         df = pli.DataFrame({"a": [1]})
-        return df.select(pli.arange(low, high, step).alias("arange"))["arange"]  # type: ignore
+        return df.select(arange(low, high, step).alias("arange"))["arange"]  # type: ignore
 
     return pli.wrap_expr(pyarange(low._pyexpr, high._pyexpr, step))