Populate init files

2020-04-11 17:35:00 +02:00 · 2020-04-11 17:35:00 +02:00 · e54bf07030
commit e54bf07030
parent 8c629c0cb1
2 changed files with 19 additions and 0 deletions
--- a/prototorch/functions/init.py
+++ b/prototorch/functions/init.py
@ -0,0 +1,12 @@
+"""ProtoTorch functions."""
+
+from .activations import identity, sigmoid_beta, swish_beta
+from .competitions import knnc, wtac
+
+__all__ = [
+    'identity',
+    'sigmoid_beta',
+    'swish_beta',
+    'knnc',
+    'wtac',
+]
--- a/prototorch/modules/init.py
+++ b/prototorch/modules/init.py
@ -0,0 +1,7 @@
+"""ProtoTorch modules."""
+
+from .prototypes import Prototypes1D
+
+__all__ = [
+    'Prototypes1D',
+]