sty: run ruff

nipreps · May 1, 2024 · 4cea98e · 4cea98e
1 parent 04849d5
commit 4cea98e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/mriqc_learn/datasets/__init__.py b/mriqc_learn/datasets/__init__.py
@@ -21,6 +21,7 @@
 #     https://www.nipreps.org/community/licensing/
 #
 """Public API for MRIQC-learn datasets."""
+
 from pathlib import Path
 from pkg_resources import resource_filename as pkgrf
 
@@ -101,7 +102,7 @@ def load_data(
     if split_strategy is None or split_strategy.lower() == "none":
         return (
             dataframe[dataframe.columns[xy_index:]],
-            dataframe[dataframe.columns[:xy_index]]
+            dataframe[dataframe.columns[:xy_index]],
         ), (None, None)
 
     n = len(dataframe)
@@ -122,8 +123,5 @@ def load_data(
 
     return (
         train_df[dataframe.columns[xy_index:]],
-        train_df[dataframe.columns[:xy_index]]
-    ), (
-        test_df[dataframe.columns[xy_index:]],
-        test_df[dataframe.columns[:xy_index]]
-    )
+        train_df[dataframe.columns[:xy_index]],
+    ), (test_df[dataframe.columns[xy_index:]], test_df[dataframe.columns[:xy_index]])
diff --git a/mriqc_learn/models/production.py b/mriqc_learn/models/production.py
@@ -21,6 +21,7 @@
 #     https://www.nipreps.org/community/licensing/
 #
 """Create a pipeline for nested cross-validation."""
+
 from pkg_resources import resource_filename as pkgrf
 
 from joblib import load