Add 'datasets' to main package __init__.py

2020-04-14 19:51:14 +02:00 · 2020-04-14 19:51:14 +02:00 · 2b82830590
commit 2b82830590
parent 553b1e1a65
1 changed files with 2 additions and 1 deletions
--- 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',
 ]