Ruff format

Author: c-salomonsen

CollaborativeCoding/dataloaders/mnist_4_9.py

@@ -43,12 +43,11 @@ def __init__(
         self.labels_path = self.mnist_path / (
             MNIST_SOURCE["train_labels"][1] if train else MNIST_SOURCE["test_labels"][1]
         )
-        
-        # Functions to map the labels from (4,9) -> (0,5) for CrossEntropyLoss to work properly. 
-        self.label_shift = lambda x: x-4
-        self.label_restore = lambda x: x+4
-        
-        
+
+        # Functions to map the labels from (4,9) -> (0,5) for CrossEntropyLoss to work properly.
+        self.label_shift = lambda x: x - 4
+        self.label_restore = lambda x: x + 4
+
     def __len__(self):
         return len(self.samples)
 

main.py

@@ -139,7 +139,7 @@ def main():
 
     for epoch in range(args.epoch):
         # Training loop start
-        print(f"Epoch: {epoch+1}/{args.epoch}")
+        print(f"Epoch: {epoch + 1}/{args.epoch}")
         trainingloss = []
         model.train()
         for x, y in tqdm(trainloader, desc="Training"):