Skip to content

Commit 0fd6608

Browse files
committed
STY: Fix style complaints
Fix style complaints.
1 parent cc66cc9 commit 0fd6608

2 files changed

Lines changed: 6 additions & 4 deletions

File tree

src/nifreeze/model/gpr.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -490,12 +490,14 @@ def __init__(
490490
bval_index: int = 3,
491491
) -> None:
492492
# Store init params for sklearn cloning
493-
self.orientation_kernel = orientation_kernel if orientation_kernel is not None else SphericalKriging()
493+
self.orientation_kernel = (
494+
orientation_kernel if orientation_kernel is not None else SphericalKriging()
495+
)
494496
self.radial_kernel = radial_kernel if radial_kernel is not None else RBF(length_scale=1.0)
495497
self.orientation_dims = tuple(orientation_dims)
496498
self.bval_index = bval_index
497499

498-
super().__init__(self.orientation_kernel, self.radial_kernel,)
500+
super().__init__(self.orientation_kernel, self.radial_kernel)
499501

500502
def get_params(self, deep: bool = True): # noqa: D401
501503
# Return only __init__ parameters (sklearn clone() relies on this)

test/test_model_gpr.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -564,8 +564,8 @@ def test_multishellkernel_column_selection(orientation_dims, bval_index, nuisanc
564564
Y = X.copy()
565565
Y[:, bval_index] *= 1.5
566566

567-
Kxy0 = k(X, X) # includes radial similarity at identical bvals
568-
Kxy1 = k(X, Y) # radial similarity with different bvals
567+
Kxy0 = k(X, X) # includes radial similarity at identical bvals
568+
Kxy1 = k(X, Y) # radial similarity with different bvals
569569

570570
assert not np.allclose(Kxy0, Kxy1)
571571

0 commit comments

Comments
 (0)