Update train_classifier_cnn.py

usnistgov · Jan 23, 2025 · aec38fe · aec38fe
1 parent 4af1f94
commit aec38fe
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/atomvision/scripts/train_classifier_cnn.py b/atomvision/scripts/train_classifier_cnn.py
@@ -37,17 +37,17 @@
 import cv2
 from skimage.measure import regionprops
 from skimage import draw
-from alignn.models.alignn import (
-    ALIGNNConfig,
-    MLPLayer,
-    ALIGNNConv,
-    EdgeGatedGraphConv,
-)
-from alignn.models.utils import RBFExpansion
 from sklearn.metrics import confusion_matrix
 try:
     import dgl
     from dgl.nn import AvgPooling
+    from alignn.models.alignn import (
+        ALIGNNConfig,
+        MLPLayer,
+        ALIGNNConv,
+        EdgeGatedGraphConv,
+    )
+    from alignn.models.utils import RBFExpansion
 except Exception:
     pass
 # import dgl