cleaning up code

tractometry · Dec 3, 2024 · 0529ff7 · 0529ff7
1 parent 3b3b645
commit 0529ff7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/afqinsight/nn/utils.py b/afqinsight/nn/utils.py
@@ -384,10 +384,10 @@ def prep_pytorch_data(dataset):
 
     Returns:
         tuple:
-            The PyTorch dataset,
-            the training data loader,
-            the test data loader,
-            and the validation data loader.
+            PyTorch dataset,
+            Training data loader,
+            Test data loader,
+            Validation data loader.
     """
     dataset.drop_target_na()
     imputer = dataset.model_fit(SimpleImputer(strategy="median"))