Merge pull request #140 from runpod/progress-patch

fix: checking for status

Author: justinmerrell

runpod/serverless/modules/rp_http.py

@@ -55,7 +55,7 @@ async def _handle_result(session, job_data, job, url_template, log_message):
         log.error(f"Error while returning job result {job['id']}: {err}")
 
     finally:
-        if url_template == JOB_DONE_URL and job.get('status', None) != 'IN_PROGRESS':
+        if url_template == JOB_DONE_URL and job_data.get('status', None) != 'IN_PROGRESS':
             job_list.remove_job(job["id"])
             log.info(f'{job["id"]} | Finished')
 

runpod/serverless/modules/rp_progress.py

@@ -9,8 +9,11 @@
 
 import aiohttp
 
+from runpod.serverless.modules.rp_logger import RunPodLogger
 from .rp_http import send_result
 
+log = RunPodLogger()
+
 async def _create_session_async():
     """
     Creates an aiohttp session.
@@ -48,6 +51,8 @@ async def main():
                 await _async_progress_update(session, job, progress)
 
         loop.run_until_complete(main())
+
+        log.debug(f'{job["id"]} | Progress Update Sent: {progress}')
     finally:
         loop.close()
 
@@ -56,5 +61,6 @@ def progress_update(job, progress):
     """
     Updates the progress of a currently running job in a separate thread.
     """
+    log.debug(f'{job["id"]} | Sending Progress Update: {progress}')
     thread = threading.Thread(target=_thread_target, args=(job, progress), daemon=True)
     thread.start()