From d834b1b4321b6defdf81c70e26a7b66eec8ba753 Mon Sep 17 00:00:00 2001 From: Peng Lu Date: Wed, 1 Feb 2023 19:29:57 +0800 Subject: [PATCH] fix datasets import bug (#1943) --- mmpose/datasets/builder.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/mmpose/datasets/builder.py b/mmpose/datasets/builder.py index b3993a7ebb..2e5a236ff4 100644 --- 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])