fix: revert stylistic changes

nipreps · Apr 5, 2024 · 5155f9f · 5155f9f
1 parent 148bb71
commit 5155f9f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/src/eddymotion/data/splitting.py b/src/eddymotion/data/splitting.py
@@ -21,11 +21,9 @@
 #     https://www.nipreps.org/community/licensing/
 #
 """Data splitting helpers."""
-
 from pathlib import Path
-
-import h5py
 import numpy as np
+import h5py
 
 
 def lovo_split(dataset, index, with_b0=False):

diff --git a/test/test_model.py b/test/test_model.py
@@ -26,8 +26,8 @@
 import pytest
 
 from eddymotion import model
-from eddymotion.data.dmri import DWI
 from eddymotion.data.splitting import lovo_split
+from eddymotion.data.dmri import DWI
 
 
 def test_trivial_model():

diff --git a/test/test_splitting.py b/test/test_splitting.py
@@ -52,7 +52,8 @@ def test_lovo_split(datadir):
     data.gradients[..., index] = 1
 
     # Apply the lovo_split function at the specified index
-    (train_data, train_gradients), (test_data, test_gradients) = lovo_split(data, index)
+    (train_data, train_gradients), \
+    (test_data, test_gradients) = lovo_split(data, index)
 
     # Check if the test data contains only 1s
     # and the train data contains only 0s after the split