fix ut

caic99 · caic99 · commit e6f81a5d530e · 2024-11-28T17:03:39.000+08:00
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):

Original file line number	Diff line number	Diff line change
`@@ -28,6 +28,7 @@`
`28`	`28`	`)`
`29`	`29`
`30`	`30`	`from deepmd.pt.utils import (`
	`31`	`+ dp_random,`
`31`	`32`	`env,`
`32`	`33`	`)`
`33`	`34`	`from deepmd.pt.utils.dataset import (`
`@@ -50,7 +51,7 @@ def setup_seed(seed) -> None:`
`50`	`51`	`torch.manual_seed(seed)`
`51`	`52`	`torch.cuda.manual_seed_all(seed)`
`52`	`53`	`torch.backends.cudnn.deterministic = True`
`53`		`- np.random.seed(seed)`
	`54`	`+ dp_random.seed(seed)`
`54`	`55`
`55`	`56`
`56`	`57`	`class DpLoaderSet(Dataset):`