Merge pull request #1276 from NASA-IMPACT/dev

Added objects for weekend schedule

Author: dhanur-sharma

inference/models/inference.py

@@ -301,7 +301,6 @@ def refresh_status_and_store_results(self) -> None:
         """Process this external job and update status/results"""
         try:
             api_client = InferenceAPIClient()
-            # model_version = ModelVersion.objects.get(classification_type=self.inference_job.classification_type)
             model_version = self.inference_job.model_version
 
             response = api_client.get_job_status(model_version.api_identifier, self.external_job_id)
@@ -314,7 +313,6 @@ def refresh_status_and_store_results(self) -> None:
             # Handle completion or failure
             if new_status == ExternalJobStatus.COMPLETED:
                 self.store_results(response.get("results"))
-                # self.completed_at = timezone.now() # completed in mark_completed called in store_results
             self.save()
 
         except Exception as e:

inference/signals.py

@@ -22,3 +22,19 @@ def create_periodic_tasks(sender, **kwargs):
             name="Process inference queue (6pm-7am)",
             task="inference.tasks.process_inference_job_queue",
         )
+
+        # Create schedule the weekend
+        weekend_crontab, _ = CrontabSchedule.objects.get_or_create(
+            minute="*/5",
+            hour="*",
+            day_of_week="0,6",
+            day_of_month="*",
+            month_of_year="*",
+        )
+
+        # Create the periodic task if it doesn't exist
+        PeriodicTask.objects.get_or_create(
+            crontab=weekend_crontab,
+            name="Process inference queue (Weekends)",
+            task="inference.tasks.process_inference_job_queue",
+        )

inference/tasks.py

@@ -18,10 +18,6 @@ def process_inference_job_queue():
             return "Queue processing already in progress"
 
         try:
-            # Reevaluate progress and update status of all inference jobs that are not currently queued
-            # for job in InferenceJob.objects.exclude(status=InferenceJobStatus.QUEUED):
-            #     job.reevaluate_progress_and_update_status()
-
             # Look for pending jobs first
             pending_jobs = InferenceJob.objects.filter(status=InferenceJobStatus.PENDING)