From de6d22b4922e05a1aa17d43b43b25415fdf8c915 Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Thu, 19 Dec 2024 04:24:23 -0500
Subject: [PATCH] fix import

---
 deepmd/pd/utils/dataloader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deepmd/pd/utils/dataloader.py b/deepmd/pd/utils/dataloader.py
index c3663ff679..0d1d6af337 100644
--- a/deepmd/pd/utils/dataloader.py
+++ b/deepmd/pd/utils/dataloader.py
@@ -40,7 +40,7 @@
     mix_entropy,
 )
 from deepmd.utils import (
-    dp_random,
+    random as dp_random,
 )
 from deepmd.utils.data import (
     DataRequirementItem,