Skip to content

Commit 61f93e4

Browse files
committed
fix lint
1 parent 3a20509 commit 61f93e4

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

tests/primitives/prompting/test_timeseries_preprocessing.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def test_rolling_window_sequences(values, window_size, step_size):
3131
np.array([0, 1, 2, 3, 4]),
3232
3,
3333
1,
34-
1
34+
1,
3535
)
3636

3737
result = rolling_window_sequences(values, window_size, step_size)
@@ -44,12 +44,7 @@ def test_rolling_window_sequences(values, window_size, step_size):
4444

4545

4646
def test_rolling_window_sequences_multivariate(window_size, step_size):
47-
values = np.array([
48-
[0.555, 2.345],
49-
[1.501, 5.903],
50-
[9.116, 3.068],
51-
[7.432, 4.532]
52-
])
47+
values = np.array([[0.555, 2.345], [1.501, 5.903], [9.116, 3.068], [7.432, 4.532]])
5348

5449
expected = (
5550
np.array([
@@ -59,7 +54,7 @@ def test_rolling_window_sequences_multivariate(window_size, step_size):
5954
np.array([0, 1]),
6055
3,
6156
1,
62-
2
57+
2,
6358
)
6459

6560
result = rolling_window_sequences(values, window_size, step_size)

0 commit comments

Comments
 (0)