for testing and deploying your application
for finding and fixing issues
for empowering human code reviews
"""Registry of backend types for Annif"""
# define functions for lazily importing each backend (alphabetical order)
def _dummy():
from . import dummy
return dummy.DummyBackend
def _ensemble():
from . import ensemble
return ensemble.EnsembleBackend
def _fasttext():
try:
from . import fasttext
return fasttext.FastTextBackend
except ImportError:
raise ValueError("fastText not available, cannot use fasttext backend")
def _http():
from . import http
return http.HTTPBackend
def _mllm():
from . import mllm
return mllm.MLLMBackend
def _nn_ensemble():
from . import nn_ensemble
return nn_ensemble.NNEnsembleBackend
raise ValueError(
"Keras and TensorFlow not available, cannot use " + "nn_ensemble backend"
)
def _omikuji():
from . import omikuji
return omikuji.OmikujiBackend
raise ValueError("Omikuji not available, cannot use omikuji backend")
def _pav():
from . import pav
return pav.PAVBackend
def _stwfsa():
from . import stwfsa
return stwfsa.StwfsaBackend
def _svc():
from . import svc
return svc.SVCBackend
def _tfidf():
from . import tfidf
return tfidf.TFIDFBackend
def _yake():
from . import yake
return yake.YakeBackend
raise ValueError("YAKE not available, cannot use yake backend")
# registry of the above functions
_backend_fns = {
"dummy": _dummy,
"ensemble": _ensemble,
"fasttext": _fasttext,
"http": _http,
"mllm": _mllm,
"nn_ensemble": _nn_ensemble,
"omikuji": _omikuji,
"pav": _pav,
"stwfsa": _stwfsa,
"svc": _svc,
"tfidf": _tfidf,
"yake": _yake,
}
def get_backend(backend_id):
if backend_id in _backend_fns:
return _backend_fns[backend_id]()
else:
raise ValueError("No such backend type {}".format(backend_id))
NatLibFi /
Annif
