fix: catching more errors

Author: Justin Merrell

runpod/serverless/modules/job.py

@@ -41,6 +41,11 @@ def get(worker_id):
                 log(f"TAKE_JOB URL response: {assigned_job.status_code}")
                 get_return = assigned_job.json()
 
+    # Status code 400
+    except requests.exceptions.HTTPError:
+        log("HTTPError while requesting job", 'WARNING')
+
+    # Status code 408
     except requests.exceptions.Timeout:
         log("Timeout while requesting job", 'WARNING')
 
@@ -113,6 +118,12 @@ def post(worker_id, job_id, job_output):
 
     try:
         rp_session.post(job_done_url, data=job_data, headers=headers, timeout=10)
+
+    # Status code 400
+    except requests.exceptions.HTTPError:
+        log(f"HTTPError while completing job {job_id}")
+
+    # Status code 408
     except requests.exceptions.Timeout:
         log(f"Timeout while completing job {job_id}")
 

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = runpod
-version = 0.6.2
+version = 0.6.3
 description = Official Python library for RunPod API & SDK.
 long_description = file: README.md
 long_description_content_type = text/markdown