removing dataset postfix

2c277d62 · soumith · aaa218de · 2c277d62
Commit 2c277d62 authored Nov 09, 2016 by soumith
Show 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 .folder import ImageFolderDataset
-from .coco import CocoCaptionsDataset, CocoDetectionDataset
+from .lsun import LSUN, LSUNClass
+from .folder import ImageFolder
+from .coco import CocoCaptions, CocoDetection

-__all__ = ('LSUNDataset', 'LSUNClassDataset',
-           'ImageFolderDataset',
-           'CocoCaptionsDataset', 'CocoDetectionDataset')
+__all__ = ('LSUN', 'LSUNClass',
+           'ImageFolder',
+           'CocoCaptions', 'CocoDetection')