Make sure all submodules are properly registered
# What does this PR do?

We have seen some mysterious bugs in Python 3.8 linked to the lazy init in Transformers, where some of the submodules are not properly registered as such because they don't appear in the `_import_structure` dict. This PR fixes that and adds a check to make sure newly added modules are always added to the init.