lint

koaning · Jul 2, 2024 · e2dc1cc · e2dc1cc
1 parent 3bd1895
commit e2dc1cc
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/tests/test_pandas_utils/test_pandas_utils.py b/tests/test_pandas_utils/test_pandas_utils.py
@@ -38,11 +38,10 @@ def test_add_lags_wrong_inputs(data, frame_func):
         add_lags(invalid_df, ["X1"], 1)
 
 
-@pytest.mark.parametrize("frame_func", [
-    pd.DataFrame,
-    lambda data: pl.DataFrame(data, strict=False),
-    lambda data: pl.LazyFrame(data, strict=False)
-])
+@pytest.mark.parametrize(
+    "frame_func",
+    [pd.DataFrame, lambda data: pl.DataFrame(data, strict=False), lambda data: pl.LazyFrame(data, strict=False)],
+)
 def test_add_lags_correct_df(data, frame_func):
     test_df = frame_func(data)
     expected = frame_func({"X1": [1, 2], "X2": ["178", "154"], "X1-1": [0, 1]})