Skip to content

Commit 0436f39

Browse files
[pre-commit.ci lite] apply automatic fixes
1 parent 2b199f3 commit 0436f39

File tree

3 files changed

+12
-9
lines changed

3 files changed

+12
-9
lines changed

tsml/transformations/_quantile.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from typing import Union, List
1+
from typing import List, Union
22

33
import numpy as np
44
from sklearn.base import TransformerMixin
@@ -19,13 +19,15 @@ def __init__(
1919

2020
super(QuantileTransformer).__init__()
2121

22-
def fit(self, X: Union[np.ndarray, List[np.ndarray]], y: Union[np.ndarray, None] =None) -> object:
22+
def fit(
23+
self, X: Union[np.ndarray, List[np.ndarray]], y: Union[np.ndarray, None] = None
24+
) -> object:
2325
"""Unused. Validates X."""
2426
self._validate_data(X=X)
2527
return self
2628

2729
def transform(
28-
self, X: Union[np.ndarray, List[np.ndarray]], y: Union[np.ndarray, None] =None
30+
self, X: Union[np.ndarray, List[np.ndarray]], y: Union[np.ndarray, None] = None
2931
) -> np.ndarray:
3032
"""Transform input cases in X.
3133
@@ -48,8 +50,8 @@ def transform(
4850
if num_quantiles == 1:
4951
return X.quantile(torch.tensor([0.5]), dim=-1).permute(1, 2, 0)
5052
else:
51-
quantiles = X.quantile(
52-
torch.linspace(0, 1, num_quantiles), dim=-1
53-
).permute(1, 2, 0)
53+
quantiles = X.quantile(torch.linspace(0, 1, num_quantiles), dim=-1).permute(
54+
1, 2, 0
55+
)
5456
quantiles[..., 1::2] = quantiles[..., 1::2] - X.mean(-1, keepdims=True)
55-
return quantiles
57+
return quantiles

tsml/utils/numba_functions/stats.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
# -*- coding: utf-8 -*-
21
"""Numba statistic utilities."""
32

43
__author__ = ["MatthewMiddlehurst"]

tsml/utils/validation.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,9 @@ def check_n_jobs(n_jobs: int) -> int:
5858
elif n_jobs < 0:
5959
cpu_count = os.cpu_count()
6060
if cpu_count is None:
61-
warnings.warn("Could not detect CPU count. Setting `n_jobs` to 1.", stacklevel=2)
61+
warnings.warn(
62+
"Could not detect CPU count. Setting `n_jobs` to 1.", stacklevel=2
63+
)
6264
return 1
6365
else:
6466
return max(1, cpu_count + 1 + n_jobs)

0 commit comments

Comments
 (0)