fix flake8 error

c47c4480 · Kai Chen · bac4c32f · c47c4480 · c47c4480
Commit c47c4480 authored Oct 07, 2018 by Kai Chen
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tools/test.py tools/test.py +1 -1

tools/train.py tools/train.py +1 -1

No files found.
--- a/tools/test.py
+++ b/tools/test.py
@@ -6,7 +6,7 @@ from mmcv.runner import load_checkpoint, parallel_test, obj_from_dict

 from mmdet import datasets
 from mmdet.core import scatter, MMDataParallel, results2json, coco_eval
-from mmdet.datasets.loader import collate, build_dataloader
+from mmdet.datasets import collate, build_dataloader
 from mmdet.models import build_detector, detectors



--- a/tools/train.py
+++ b/tools/train.py
@@ -13,7 +13,7 @@ from mmdet import datasets, __version__
 from mmdet.core import (init_dist, DistOptimizerHook, DistSamplerSeedHook,
                        MMDataParallel, MMDistributedDataParallel,
                        CocoDistEvalRecallHook, CocoDistEvalmAPHook)
-from mmdet.datasets.loader import build_dataloader
+from mmdet.datasets import build_dataloader
 from mmdet.models import build_detector, RPN