From 17315ff24253fc66906b7eadb00b20bdb2ca8e94 Mon Sep 17 00:00:00 2001
From: Jensun Ravichandran <jjensun@gmail.com>
Date: Fri, 7 May 2021 15:23:52 +0200
Subject: [PATCH] Add models to the prototorch.models namespace

---
 prototorch/models/__init__.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/prototorch/models/__init__.py b/prototorch/models/__init__.py
index 2507e45..bde1ef1 100644
--- a/prototorch/models/__init__.py
+++ b/prototorch/models/__init__.py
@@ -1,5 +1,10 @@
 from importlib.metadata import PackageNotFoundError, version
 
+from .cbc import CBC
+from .glvq import GLVQ, GMLVQ, GRLVQ, LVQMLN, ImageGLVQ, SiameseGLVQ
+from .neural_gas import NeuralGas
+from .vis import *
+
 VERSION_FALLBACK = "uninstalled_version"
 try:
     __version__ = version(__name__.replace(".", "-"))