Update models namespace

2021-05-21 13:11:59 +02:00 · 2021-05-21 13:11:59 +02:00 · 0611f81aba
commit 0611f81aba
parent a9382dcd9b
1 changed files with 2 additions and 2 deletions
--- a/prototorch/models/init.py
+++ b/prototorch/models/init.py
@ -1,7 +1,7 @@
 from importlib.metadata import PackageNotFoundError, version

-from .cbc import CBC
-from .glvq import (GLVQ, GMLVQ, GRLVQ, GLVQ1, GLVQ21, LVQ1, LVQ21, LVQMLN,
+from .cbc import CBC, ImageCBC
+from .glvq import (GLVQ, GLVQ1, GLVQ21, GMLVQ, GRLVQ, LVQ1, LVQ21, LVQMLN,
                   ImageGLVQ, ImageGMLVQ, SiameseGLVQ)
 from .knn import KNN
 from .neural_gas import NeuralGas