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

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

training/src/anemoi/training/train/tasks/base.py

@@ -748,7 +748,7 @@ def _normalize_batch(self, batch: torch.Tensor) -> torch.Tensor:
             Normalized batch
         """
         for dataset_name in batch:
-            batch[dataset_name] = self.model.pre_processors[dataset_name](batch[dataset_name]) # normalized in-place
+            batch[dataset_name] = self.model.pre_processors[dataset_name](batch[dataset_name])  # normalized in-place
         return batch
 
     def _prepare_loss_scalers(self) -> None:

training/src/anemoi/training/train/train.py

@@ -54,13 +54,14 @@ def instanciate_wrapper(config: Any, *args, **kwargs) -> Any:
             target = "<no _target_>"
         try:
             return hydra_instantiate(config, *args, **kwargs)
-        except Exception as e:  # noqa
+        except Exception as e:
             config_keys = str(target) + "+" + config_keys
+
             def show_if_dict(v):
-                if hasattr(v, "keys"): # must be a dict
+                if hasattr(v, "keys"):  # must be a dict
                     return v.__class__.__name__ + f"({'+'.join(v.keys())})"
-                else:
-                    return v.__class__.__name__
+                return v.__class__.__name__
+
             _args = ",".join(show_if_dict(a) for a in args)
             _kwargs = ",".join(f"{k}={show_if_dict(v)}" for k, v in kwargs.items())
             e.add_note(f"This exception happend when doing instanciate for config: {config}")