diff --git a/scripts_python/ae_train_model.py b/scripts_python/ae_train_model.py
index 6bf9aec9..e5846235 100755
--- a/scripts_python/ae_train_model.py
+++ b/scripts_python/ae_train_model.py
@@ -11,7 +11,7 @@
 # comet_ml not used, but comet_ml requires to be imported before torch.
 # See bug report here https://github.com/Lightning-AI/lightning/issues/5829
 # Importing now to solve issues later.
-import comet_ml # noqa F401
+import comet_ml  # noqa F401
 import torch
 
 from scilpy.io.utils import (assert_inputs_exist, assert_outputs_exist,