Refactor imports

uktrade · Mar 12, 2024 · ebbe2f5 · ebbe2f5
1 parent 5580475
commit ebbe2f5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/app/views.py b/app/views.py
@@ -14,7 +14,7 @@
 
 from celery_worker.tasks import demodjango_task
 from .check.check_http import HTTPCheck
-from .models import ScheduledTask
+
 from .util import render_connection_info
 
 logger = logging.getLogger("django")
@@ -198,6 +198,7 @@ def get_result_from_celery_backend():
 
 
 def celery_beat_check():
+    from .models import ScheduledTask
     addon_type = ALL_CHECKS[BEAT]
 
     try:

diff --git a/celery_worker/tasks.py b/celery_worker/tasks.py
@@ -3,7 +3,6 @@
 
 from celery import shared_task
 
-from app.models import ScheduledTask
 
 logger = logging.getLogger("django")
 
@@ -16,6 +15,8 @@ def demodjango_task(timestamp):
 
 @shared_task(bind=True)
 def demodjango_scheduled_task(self):
+    from app.models import ScheduledTask
+
     timestamp = datetime.utcnow()
 
     task = ScheduledTask()