diff --git a/test/test_splitting.py b/test/test_splitting.py index 8718964a..695283a3 100644 --- a/test/test_splitting.py +++ b/test/test_splitting.py @@ -21,9 +21,7 @@ # 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 @@ -53,7 +51,7 @@ def test_lovo_split(datadir): # Apply the lovo_split function at the specified index (train_data, train_gradients), \ - (test_data, test_gradients) = lovo_split(data, index) + (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