from wagtail import hooks

from .views import contact_entity_chooser_viewset


@hooks.register("register_admin_viewset")
def register_viewset():
    return contact_entity_chooser_viewset