Merge pull request #184 from EmmaRenauld/small_fix

Adding back auto-cleaned line. oups.
scil-vital · Jul 3, 2023 · 31476d3 · 31476d3
2 parents 82fab85 + caf1c1c
commit 31476d3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/scripts_python/l2t_train_model.py b/scripts_python/l2t_train_model.py
@@ -11,6 +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
 import torch
 
 from scilpy.io.utils import assert_inputs_exist, assert_outputs_exist