Add __all__ in metalearners

tristandeleu · Oct 9, 2019 · 6b8336c · 6b8336c
1 parent cb9ef3a
commit 6b8336c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/maml/metalearners/__init__.py b/maml/metalearners/__init__.py
@@ -1,2 +1,4 @@
 from maml.metalearners.maml import ModelAgnosticMetaLearning, MAML, FOMAML
-from maml.metalearners.meta_sgd import MetaSGD
+from maml.metalearners.meta_sgd import MetaSGD
+
+__all__ = ['ModelAgnosticMetaLearning', 'MAML', 'FOMAML', 'MetaSGD']