- more imports in init files

inferno-pytorch · Sep 1, 2017 · 09c692d · 09c692d
1 parent 38266cc
commit 09c692d
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/inferno/extensions/__init__.py b/inferno/extensions/__init__.py
@@ -1 +1,6 @@
-from . import *
+from . import containers
+from . import criteria
+from . import initializers
+from . import layers
+from . import metrics
+from . import optimizers
diff --git a/inferno/trainers/__init__.py b/inferno/trainers/__init__.py
@@ -0,0 +1,2 @@
+from . import basic
+from . import callbacks
diff --git a/inferno/trainers/callbacks/__init__.py b/inferno/trainers/callbacks/__init__.py
@@ -1 +1,3 @@
-from .base import CallbackEngine
+from .base import CallbackEngine, Callback
+from . import essentials
+from . import scheduling