Format training utils

1 files changed, 3 insertions, 1 deletions

diff --git a/training/utils.py b/training/utils.py
index 41e0a71..1996f0a 100644
--- a/training/utils.py
+++ b/training/utils.py
@@ -38,6 +38,7 @@ def configure_logging(config: DictConfig) -> None:
 
 def configure_callbacks(config: DictConfig,) -> List[Type[Callback]]:
     """Configures Lightning callbacks."""
+
     def load_callback(callback_config: DictConfig) -> Type[Callback]:
         log.info(f"Instantiating callback <{callback_config._target_}>")
         return hydra.utils.instantiate(callback_config)
@@ -47,7 +48,7 @@ def configure_callbacks(config: DictConfig,) -> List[Type[Callback]]:
         for callback_config in callback_configs.values():
             if callback_config.get("_target_"):
                 callbacks.append(load_callback(callback_config))
-            else: 
+            else:
                 callbacks += load_callbacks(callback_config)
         return callbacks
 
@@ -58,6 +59,7 @@ def configure_callbacks(config: DictConfig,) -> List[Type[Callback]]:
 
 def configure_logger(config: DictConfig) -> List[Type[LightningLoggerBase]]:
     """Configures Lightning loggers."""
+
     def load_logger(logger_config: DictConfig) -> Type[LightningLoggerBase]:
         log.info(f"Instantiating logger <{logger_config._target_}>")
         return hydra.utils.instantiate(logger_config)