fix ut

deepmodeling · Nov 28, 2024 · ddc39de · ddc39de
1 parent 3895c94
commit ddc39de
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepmd/pt/utils/dataloader.py b/deepmd/pt/utils/dataloader.py
@@ -222,7 +222,7 @@ def run(self) -> None:
             self._queue.put(item)  # Blocking if the queue is full
 
         # Signal the consumer we are done; this should not happen for DataLoader
-        self._queue.put(StopIteration)
+        self._queue.put(StopIteration())
 
 
 QUEUESIZE = 32
@@ -250,7 +250,7 @@ def __next__(self):
             wait_time > 1.0
         ):  # Even for Multi-Task training, each step usually takes < 1s
             log.warning(f"Data loading is slow, waited {wait_time:.2f} seconds.")
-        if issubclass(item, Exception):
+        if isinstance(item, Exception):
             raise item
         return item