change init file in datasets. (#93)

0885ab46 · yinchimaoliang · GitHub · b28cbf71 · 0885ab46
Unverified Commit 0885ab46 authored Sep 08, 2020 by yinchimaoliang Committed by GitHub Sep 08, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

mmdet3d/datasets/__init__.py mmdet3d/datasets/__init__.py +2 -1

No files found.
--- a/mmdet3d/datasets/__init__.py
+++ b/mmdet3d/datasets/__init__.py
-from mmdet.datasets.builder import DATASETS, build_dataloader, build_dataset
+from mmdet.datasets.builder import build_dataloader
+from .builder import DATASETS, build_dataset
 from .custom_3d import Custom3DDataset
 from .kitti_dataset import KittiDataset
 from .lyft_dataset import LyftDataset