@@ -113,11 +113,11 @@ def update(self, epoch: int, state_dict: dict, last: bool = False) -> None:
         ):
             current_best = self.best()
             self._best_metric = metrics[self._key]
-            
+
             # Save the new best model
             save_path = self.snapshot_path(epoch, best=True)
             parsed_state_dict = {
-                k: v 
+                k: v
                 for k, v in state_dict.items()
                 if self.save_optimizer_state or k != "optimizer"
             }