fix ut

deepmodeling · Nov 28, 2024 · e6f81a5 · e6f81a5
1 parent e0ace97
commit e6f81a5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deepmd/pt/utils/dataloader.py b/deepmd/pt/utils/dataloader.py
@@ -28,6 +28,7 @@
 )
 
 from deepmd.pt.utils import (
+    dp_random,
     env,
 )
 from deepmd.pt.utils.dataset import (
@@ -50,7 +51,7 @@ def setup_seed(seed) -> None:
     torch.manual_seed(seed)
     torch.cuda.manual_seed_all(seed)
     torch.backends.cudnn.deterministic = True
-    np.random.seed(seed)
+    dp_random.seed(seed)
 
 
 class DpLoaderSet(Dataset):