Remove unused imports
Don't import tests in import_all_models