- more alls

inferno-pytorch · Sep 1, 2017 · d989682 · d989682
1 parent fb0d178
commit d989682
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/inferno/extensions/__init__.py b/inferno/extensions/__init__.py
@@ -4,3 +4,5 @@
 from . import layers
 from . import metrics
 from . import optimizers
+
+__all__ = ['containers', 'criteria', 'initializers', 'layers', 'metrics', 'optimizers']
diff --git a/inferno/extensions/containers/__init__.py b/inferno/extensions/containers/__init__.py
@@ -1,2 +1,2 @@
-from .graph import Graph
-from .sequential import Sequential1, Sequential2
+from .graph import *
+from .sequential import *