fixing code format

securefederatedai · Dec 11, 2024 · 2c731c1 · 2c731c1
1 parent 275fb37
commit 2c731c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openfl-workspace/torch_cnn_mnist/src/dataloader.py b/openfl-workspace/torch_cnn_mnist/src/dataloader.py
@@ -135,4 +135,4 @@ def _load_raw_datashards(shard_num, collaborator_count, transform=None):
     X_valid = X_valid_tot[shard_num::collaborator_count].unsqueeze(1).float()
     y_valid = y_valid_tot[shard_num::collaborator_count]
 
-    return (X_train, y_train), (X_valid, y_valid)
+    return (X_train, y_train), (X_valid, y_valid)