[Fix] mmcls import error (#38)

open-mmlab · Jan 19, 2022 · 555a348 · 555a348
1 parent 6b566e1
commit 555a348
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmfewshot/classification/datasets/builder.py b/mmfewshot/classification/datasets/builder.py
@@ -4,7 +4,8 @@
 from typing import Dict, Optional
 
 from mmcls.datasets import ClassBalancedDataset, ConcatDataset, RepeatDataset
-from mmcls.datasets.builder import DATASETS, DistributedSampler, worker_init_fn
+from mmcls.datasets.builder import DATASETS, worker_init_fn
+from mmcls.datasets.samplers import DistributedSampler
 from mmcv.runner import get_dist_info
 from mmcv.utils import build_from_cfg
 from torch.utils.data import DataLoader, Dataset