Merge pull request #154 from esavary/style-correction

STY: fix style errors

src/eddymotion/data/splitting.py

@@ -21,9 +21,11 @@
 #     https://www.nipreps.org/community/licensing/
 #
 """Data splitting helpers."""
+
 from pathlib import Path
-import numpy as np
+
 import h5py
+import numpy as np
 
 
 def lovo_split(dataset, index, with_b0=False):

test/test_model.py

@@ -26,8 +26,8 @@
 import pytest
 
 from eddymotion import model
-from eddymotion.data.splitting import lovo_split
 from eddymotion.data.dmri import DWI
+from eddymotion.data.splitting import lovo_split
 
 
 def test_trivial_model():

test/test_splitting.py

@@ -21,7 +21,9 @@
 #     https://www.nipreps.org/community/licensing/
 #
 """Unit test testing the lovo_split function."""
+
 import numpy as np
+
 from eddymotion.data.dmri import DWI
 from eddymotion.data.splitting import lovo_split
 
@@ -50,13 +52,11 @@ 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
     assert np.all(test_data == 1)
     assert np.all(train_data == 0)
     assert np.all(test_gradients == 1)
     assert np.all(train_gradients == 0)
-