Skip to content

Commit b428cee

Browse files
committed
Apply black code formatting and fix missing config imports for style compliance
1 parent c683fbb commit b428cee

12 files changed

Lines changed: 14 additions & 11 deletions

stumpy/config.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,6 @@
4343
RNG = np.random.default_rng()
4444

4545

46-
4746
def _reset(var=None):
4847
"""
4948
Reset the value of a configuration variable(s) to their default value(s)

tests/test_aamp_mmotifs.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,9 +51,7 @@
5151

5252

5353
def test_aamp_mmotifs_default_parameters():
54-
motif_distances_ref = np.array(
55-
[[0.0, 0.07497313, 0.35610208]]
56-
)
54+
motif_distances_ref = np.array([[0.0, 0.07497313, 0.35610208]])
5755
motif_indices_ref = np.array([[17, 57, 25]])
5856
motif_subspaces_ref = [np.array([1])]
5957
motif_mdls_ref = [

tests/test_cache.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
import numba
2-
import numpy as np
32

43
from stumpy import cache, config
54
from stumpy.stump import stump

tests/test_gpu_aamp_ostinato.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
import naive
1313
import pytest
1414

15-
from stumpy import core
15+
from stumpy import core, config
1616

1717
if cuda.is_available():
1818
from stumpy.gpu_aamp_ostinato import gpu_aamp_ostinato

tests/test_gpu_aamp_stimp.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from unittest.mock import patch
22

33
import numpy as np
4+
from stumpy import config
45
import numpy.testing as npt
56
from numba import cuda
67

tests/test_gpu_aampdist.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from unittest.mock import patch
22

33
import numpy as np
4+
from stumpy import config
45
import numpy.testing as npt
56
from numba import cuda
67

tests/test_gpu_mpdist.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from unittest.mock import patch
33

44
import numpy as np
5+
from stumpy import config
56
import numpy.testing as npt
67
from numba import cuda
78

tests/test_gpu_ostinato.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
import naive
1515
import pytest
1616

17-
from stumpy import core
17+
from stumpy import core, config
1818

1919
if cuda.is_available():
2020
from stumpy.gpu_ostinato import gpu_ostinato

tests/test_gpu_stimp.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from unittest.mock import patch
33

44
import numpy as np
5+
from stumpy import config
56
import numpy.testing as npt
67
from numba import cuda
78

tests/test_motifs.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -228,8 +228,12 @@ def test_motifs_two_motifs():
228228
[0.0, 0.0, np.nan],
229229
[
230230
0.0,
231-
naive.distance(core.z_norm(T[idx[0] : idx[0] + m]), core.z_norm(T[idx[1] : idx[1] + m])),
232-
naive.distance(core.z_norm(T[idx[0] : idx[0] + m]), core.z_norm(T[idx[2] : idx[2] + m])),
231+
naive.distance(
232+
core.z_norm(T[idx[0] : idx[0] + m]), core.z_norm(T[idx[1] : idx[1] + m])
233+
),
234+
naive.distance(
235+
core.z_norm(T[idx[0] : idx[0] + m]), core.z_norm(T[idx[2] : idx[2] + m])
236+
),
233237
],
234238
]
235239

0 commit comments

Comments
 (0)