Import VAEModel and AVA in src/vak/models/__init__.py

vocalpy · Sep 29, 2023 · 1680dc0 · 1680dc0
1 parent f681d51
commit 1680dc0
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/vak/models/__init__.py b/src/vak/models/__init__.py
@@ -8,6 +8,8 @@
 from .parametric_umap_model import ParametricUMAPModel
 from .registry import model_family
 from .tweetynet import TweetyNet
+from .vae_model import VAEModel
+from .ava import AVA
 
 __all__ = [
     "base",
@@ -23,4 +25,7 @@
     "ParametricUMAPModel",
     "registry",
     "TweetyNet",
+    "VAEModel",
+    "AVA",
+
 ]