import DeepWFC

caic99 · Nov 22, 2024 · 3ff6cc2 · 3ff6cc2
1 parent df7121d
commit 3ff6cc2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/deepmd/entrypoints/test.py b/deepmd/entrypoints/test.py
@@ -28,6 +28,9 @@
     DeepGlobalPolar,
     DeepPolar,
 )
+from deepmd.infer.deep_wfc import (
+    DeepWFC,
+)
 from deepmd.infer.deep_pot import (
     DeepPot,
 )
@@ -46,9 +49,6 @@
     from deepmd.infer.deep_tensor import (
         DeepTensor,
     )
-    from deepmd.infer.deep_wfc import (
-        DeepWFC,
-    )
 
 __all__ = ["test"]