lint

polars_xdt/functions.py

@@ -1,9 +1,6 @@
 from __future__ import annotations
 
-import re
 import sys
-import warnings
-from datetime import date, timedelta
 from pathlib import Path
 from typing import TYPE_CHECKING, Literal, Sequence
 
@@ -17,6 +14,8 @@
     from typing_extensions import TypeAlias
 
 if TYPE_CHECKING:
+    from datetime import date
+
     from polars.type_aliases import IntoExpr
 
 RollStrategy: TypeAlias = Literal["raise", "forward", "backward"]
@@ -50,7 +49,6 @@ def get_weekmask(weekend: Sequence[str]) -> list[bool]:
     return weekmask
 
 
-
 def is_workday(
     expr: IntoExpr,
     *,
@@ -104,7 +102,10 @@ def is_workday(
     expr = parse_into_expr(expr)
     weekend_int = [mapping[x] for x in weekend]
     if holidays is not None:
-        return ~(expr.dt.date().is_in(holidays) | expr.dt.weekday().is_in(weekend_int))
+        return ~(
+            expr.dt.date().is_in(holidays)
+            | expr.dt.weekday().is_in(weekend_int)
+        )
     return ~expr.dt.weekday().is_in(weekend_int)
 
 
@@ -675,4 +676,3 @@ def arg_previous_greater(expr: IntoExpr) -> pl.Expr:
         is_elementwise=False,
         args=[expr],
     )
-

polars_xdt/ranges.py

@@ -41,7 +41,7 @@ def date_range(
 
 @overload
 def date_range(
-    start: date|  IntoExprColumn,
+    start: date | IntoExprColumn,
     end: date | IntoExprColumn,
     interval: str | timedelta = "1d",
     *,
@@ -119,6 +119,7 @@ def date_range(  # noqa: PLR0913
     │ 2023-01-09 │
     │ 2023-01-10 │
     └────────────┘
+
     """
     if weekend == ("Sat", "Sun"):
         weekend_int = [6, 7]

src/expressions.rs

@@ -54,15 +54,13 @@ pub fn from_local_datetime_output(
     Ok(Field::new(&field.name, dtype))
 }
 
-
 #[polars_expr(output_type=Int32)]
 fn month_delta(inputs: &[Series]) -> PolarsResult<Series> {
     let start_dates = &inputs[0];
     let end_dates = &inputs[1];
     impl_month_delta(start_dates, end_dates)
 }
 
-
 #[polars_expr(output_type_func=to_local_datetime_output)]
 fn to_local_datetime(inputs: &[Series]) -> PolarsResult<Series> {
     let s1 = &inputs[0];