diff --git a/runpod/serverless/modules/rp_http.py b/runpod/serverless/modules/rp_http.py index f051d2e4..213fd42b 100644 --- a/runpod/serverless/modules/rp_http.py +++ b/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') diff --git a/runpod/serverless/modules/rp_progress.py b/runpod/serverless/modules/rp_progress.py index dff370f9..3358990d 100644 --- a/runpod/serverless/modules/rp_progress.py +++ b/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()