[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
neuronets · Apr 3, 2024 · 7a954e3 · 7a954e3
1 parent 42f7d3d
commit 7a954e3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/nobrainer/processing/generation.py b/nobrainer/processing/generation.py
@@ -152,12 +152,12 @@ def _compile():
                 num_parallel_calls=num_parallel_calls,
                 volume_shape=(resolution, resolution, resolution),
                 n_classes=1,
-                scalar_labels=True
+                scalar_labels=True,
             )
             n_epochs = info.get("epochs") or epochs
-            dataset.batch(batch_size) \
-            .normalize(info.get("normalizer") or normalizer) \
-            .repeat(n_epochs)
+            dataset.batch(batch_size).normalize(
+                info.get("normalizer") or normalizer
+            ).repeat(n_epochs)
 
             with self.strategy.scope():
                 # grow the networks by one (2^x) resolution