Ideally `__init__.py` should contain nothing but imports. So I would rather move this function to a dedicated file.