diff --git a/awx/main/models/unified_jobs.py b/awx/main/models/unified_jobs.py index f38b5e5409..ba9d51258e 100644 --- a/awx/main/models/unified_jobs.py +++ b/awx/main/models/unified_jobs.py @@ -30,7 +30,12 @@ from rest_framework.exceptions import ParseError from polymorphic.models import PolymorphicModel # AWX -from awx.main.models.base import * # noqa +from awx.main.models.base import ( + CommonModelNameNotUnique, + PasswordFieldsModel, + NotificationFieldsModel, + prevent_search +) from awx.main.dispatch.control import Control as ControlDispatcher from awx.main.models.mixins import ResourceMixin, TaskManagerUnifiedJobMixin from awx.main.utils import ( @@ -315,6 +320,7 @@ class UnifiedJobTemplate(PolymorphicModel, CommonModelNameNotUnique, Notificatio Return notification_templates relevant to this Unified Job Template ''' # NOTE: Derived classes should implement + from awx.main.models.notifications import NotificationTemplate return NotificationTemplate.objects.none() def create_unified_job(self, **kwargs):