more fixes to imports

amlrt_project/models/model_loader.py

@@ -1,9 +1,8 @@
 import logging
 from typing import Any, Dict, Optional, Type, Union
 
-from amlrt_project.models.factory import (AdamFactory, OptimFactory,
-                                          PlateauFactory, SchedulerFactory,
-                                          SGDFactory, WarmupDecayFactory)
+from amlrt_project.models.factory import SGDFactory, AdamFactory, PlateauFactory, \
+    WarmupDecayFactory, OptimFactory, SchedulerFactory
 from amlrt_project.models.my_model import SimpleMLP
 
 logger = logging.getLogger(__name__)

amlrt_project/models/my_model.py

@@ -4,9 +4,8 @@
 import pytorch_lightning as pl
 from torch import FloatTensor, LongTensor, nn
 
-from amlrt_project.models.factory import (OptimFactory,
-                                          OptimizerConfigurationFactory,
-                                          SchedulerFactory)
+from amlrt_project.models.factory import OptimFactory, OptimizerConfigurationFactory, \
+    SchedulerFactory
 from amlrt_project.models.optim import load_loss
 from amlrt_project.utils.hp_utils import check_and_log_hp
 

amlrt_project/train.py

@@ -17,9 +17,8 @@
     add_config_file_params_to_argparser, load_configs, save_hparams)
 from amlrt_project.utils.file_utils import rsync_folder
 from amlrt_project.utils.hp_utils import check_and_log_hp
-from amlrt_project.utils.logging_utils import (load_experiment_loggers,
-                                               log_exp_details,
-                                               log_hyper_parameters)
+from amlrt_project.utils.logging_utils import load_experiment_loggers, log_exp_details, \
+    log_hyper_parameters
 from amlrt_project.utils.reproducibility_utils import set_seed
 
 logger = logging.getLogger(__name__)

hooks/pre-commit

@@ -6,7 +6,6 @@ set -e
 # linting
 flake8 --ignore D .  # Check everything but docstrings
 flake8 --select D --exclude tests/  # Check only the docstrings
-isort --check .  # Check imports
 
 # Raise error if any staged notebooks contain outputs
 GITDIR=$(git rev-parse --show-toplevel) # Full path to git working directory