From 3afcf3cd49661c466c75ea536b0b2a7ff57f9a05 Mon Sep 17 00:00:00 2001
From: bbowles-personal <44647062+bbowles-personal@users.noreply.github.com>
Date: Mon, 29 Apr 2019 11:02:11 -0700
Subject: [PATCH] Expose VisionDataSet class so it can be modified (#876)

---
 torchvision/datasets/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/torchvision/datasets/__init__.py b/torchvision/datasets/__init__.py
index 8cef31b1c97..84ab280c5cd 100644
--- a/torchvision/datasets/__init__.py
+++ b/torchvision/datasets/__init__.py
@@ -17,6 +17,7 @@
 from .caltech import Caltech101, Caltech256
 from .celeba import CelebA
 from .sbd import SBDataset
+from .vision import VisionDataset
 
 __all__ = ('LSUN', 'LSUNClass',
            'ImageFolder', 'DatasetFolder', 'FakeData',