diff --git a/agnocomplete/register.py b/agnocomplete/register.py index 702deff..91ac0d0 100644 --- a/agnocomplete/register.py +++ b/agnocomplete/register.py @@ -11,6 +11,7 @@ def register(klass): "Register a class into the agnocomplete registry." logger.info("registering {}".format(klass.__name__)) AGNOCOMPLETE_REGISTRY[klass.slug] = klass + return klass def get_agnocomplete_registry():