diff --git a/deepmd/pt/utils/stat.py b/deepmd/pt/utils/stat.py index d246708e48..16c300b13a 100644 --- a/deepmd/pt/utils/stat.py +++ b/deepmd/pt/utils/stat.py @@ -8,10 +8,6 @@ Optional, Union, ) - -import numpy as np -import torch - from deepmd.pt.utils import ( AtomExcludeMask, )