fix datasets import bug (open-mmlab#1943)

ly015 · Feb 21, 2023 · d834b1b · d834b1b
1 parent 2fc9cdd
commit d834b1b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/mmpose/datasets/builder.py b/mmpose/datasets/builder.py
@@ -6,7 +6,7 @@
 import numpy as np
 import torch
 from mmengine import build_from_cfg, is_seq_of
-from torch.utils.data.dataset import ConcatDataset
+from mmengine.dataset import ConcatDataset, RepeatDataset
 
 from mmpose.registry import DATASETS
 
@@ -64,7 +64,6 @@ def build_dataset(cfg, default_args=None):
     Returns:
         Dataset: The constructed dataset.
     """
-    from .dataset_wrappers import RepeatDataset
 
     if isinstance(cfg, (list, tuple)):
         dataset = ConcatDataset([build_dataset(c, default_args) for c in cfg])