From 4954f06e0204e87d220c6b79ef1038b5d7dcea32 Mon Sep 17 00:00:00 2001
From: caic99 <amoycaic@gmail.com>
Date: Thu, 19 Dec 2024 03:09:27 +0000
Subject: [PATCH] revert changes on mp

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

diff --git a/deepmd/pt/utils/dataloader.py b/deepmd/pt/utils/dataloader.py
index 7a3cfdfc2f..8070f413e0 100644
--- a/deepmd/pt/utils/dataloader.py
+++ b/deepmd/pt/utils/dataloader.py
@@ -2,7 +2,7 @@
 import logging
 import os
 import time
-from multiprocessing import (
+from multiprocessing.dummy import (
     Pool,
 )
 from queue import (