Force execution of __init__.py on direct execution (#33)

src/nn_template/__init__.py

@@ -6,8 +6,6 @@
 # thus the logging configuration defined in the __init__.py must be called before
 # the lightning import otherwise it has no effect.
 # See https://github.com/PyTorchLightning/pytorch-lightning/issues/1503
-#
-# Force the execution of __init__.py if this file is executed directly.
 lightning_logger = logging.getLogger("pytorch_lightning")
 # Remove all handlers associated with the lightning logger.
 for handler in lightning_logger.handlers[:]:

src/nn_template/run.py

@@ -16,6 +16,9 @@
 from nn_core.model_logging import NNLogger
 from nn_core.resume import resolve_ckpt, resolve_run_path, resolve_run_version
 
+# Force the execution of __init__.py if this file is executed directly.
+import nn_template  # noqa
+
 pylogger = logging.getLogger(__name__)
 
 RESUME_MODES = {