lint

koaning · Jun 26, 2024 · 22d0489 · 22d0489
1 parent 217b1cb
commit 22d0489
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/tests/test_pandas_utils/test_pandas_utils.py b/tests/test_pandas_utils/test_pandas_utils.py
@@ -27,10 +27,7 @@ def test_X():
     return np.array([[-4, 2], [-2, 0], [4, -6]])
 
 
-@pytest.mark.parametrize("frame_func", [
-    pd.DataFrame,
-    lambda data: pl.DataFrame(data, strict=False)
-])
+@pytest.mark.parametrize("frame_func", [pd.DataFrame, lambda data: pl.DataFrame(data, strict=False)])
 def test_add_lags_wrong_inputs(data, frame_func):
     invalid_df = [[1, 2, 3], [4, 5, 6]]
     invalid_lags = ["1", "2"]