From 3ff6cc2a53f58e8ce63b3ad4a84177c0fcf2c45b Mon Sep 17 00:00:00 2001 From: Chun Cai Date: Fri, 22 Nov 2024 16:40:03 +0800 Subject: [PATCH] import `DeepWFC` --- deepmd/entrypoints/test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/deepmd/entrypoints/test.py b/deepmd/entrypoints/test.py index 3c7b4ad1c6..725cb981b8 100644 --- 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"]