removing dataset postfix

2c277d62 · soumith · aaa218de · 2c277d62
Commit 2c277d62 authored Nov 09, 2016 by soumith
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 6 deletions

torchvision/datasets/__init__.py torchvision/datasets/__init__.py +6 -6

No files found.
--- a/torchvision/datasets/__init__.py
+++ b/torchvision/datasets/__init__.py
-from .lsun import LSUNDataset, LSUNClassDataset
+from .lsun import LSUN, LSUNClass
-from .folder import ImageFolderDataset
+from .folder import ImageFolder
-from .coco import CocoCaptionsDataset, CocoDetectionDataset
+from .coco import CocoCaptions, CocoDetection
-__all__ = ('LSUNDataset', 'LSUNClassDataset',
+__all__ = ('LSUN', 'LSUNClass',
-           'ImageFolderDataset',
+           'ImageFolder',
-           'CocoCaptionsDataset', 'CocoDetectionDataset')
+           'CocoCaptions', 'CocoDetection')