fix import

AI4SIM · Jan 29, 2025 · 7c066a8 · 7c066a8
1 parent 26d8915
commit 7c066a8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/reactive-flows/cnf-combustion/gnns/tests/test_models.py b/reactive-flows/cnf-combustion/gnns/tests/test_models.py
@@ -23,9 +23,9 @@
 import torch_geometric as pyg
 import torch_optimizer as optim
 import yaml
-
 from lightning.pytorch.trainer import Trainer
-from models import LitGCN, LitGAT, LitGIN, LitGraphUNet
+
+from models import LitGAT, LitGCN, LitGIN, LitGraphUNet
 
 
 class TestModel(unittest.TestCase):