From 2b828305905a9aa7be04f372cdf15cc9950c194e Mon Sep 17 00:00:00 2001
From: blackfly <jjensun@gmail.com>
Date: Tue, 14 Apr 2020 19:51:14 +0200
Subject: [PATCH] Add 'datasets' to main package __init__.py

---
 prototorch/__init__.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/prototorch/__init__.py b/prototorch/__init__.py
index 4953edb..7ba09a0 100644
--- a/prototorch/__init__.py
+++ b/prototorch/__init__.py
@@ -2,9 +2,10 @@
 
 __version__ = '0.1.1-dev0'
 
-from prototorch import functions, modules
+from prototorch import datasets, functions, modules
 
 __all__ = [
+    'datasets',
     'functions',
     'modules',
 ]