fix: clean logs

Author: justinmerrell

runpod/serverless/modules/job.py

@@ -107,7 +107,6 @@ def run_job(handler: Callable, job: Dict[str, Any]) -> Dict[str, Any]:
         run_result = {"error": f"handler: {str(err)} \ntraceback: {traceback.format_exc()}"}
     finally:
         log.debug(f'{job["id"]} | Run result: {run_result}')
-        log.info(f'{job["id"]} | Finished')
 
         return run_result  # pylint: disable=lost-exception
 

runpod/serverless/work_loop.py

@@ -72,21 +72,22 @@ async def start_worker(config: Dict[str, Any]) -> None:
 
             # If refresh_worker is set, pod will be reset after job is complete.
             if config.get("refresh_worker", False):
-                log.info(f"Refresh worker flag set, stopping pod after job {job['id']}.")
+                log.info(f"refresh_worker | Flag set, stopping pod after job {job['id']}.")
                 job_result["stopPod"] = True
 
             if config["rp_args"].get("rp_debugger", False):
-                log.debug("rp_debugger flag set, return debugger output.")
+                log.debug("rp_debugger | Flag set, return debugger output.")
                 job_result["output"]["rp_debugger"] = rp_debugger.get_debugger_output()
 
                 ready_delay = (config["reference_counter_start"] - REF_COUNT_ZERO) * 1000
                 job_result["output"]["rp_debugger"]["ready_delay_ms"] = ready_delay
             else:
-                log.debug("rp_debugger flag not set, skipping debugger output.")
+                log.debug("rp_debugger | Flag not set, skipping debugger output.")
                 rp_debugger.clear_debugger_output()
 
             await send_result(session, job_result, job)
 
+            log.info(f'{job["id"]} | Finished')
             set_job_id(None)
 
             if _is_local_testing():