comfy runner: catch requests exceptions appropriately

This commit is contained in:
Kasisnu
2024-08-09 18:37:57 +05:30
parent 24f7e6ebcb
commit d190b34895
@@ -92,7 +92,7 @@ args = parse_args()
def get_queued_jobs_count(server_address):
try:
response = requests.get(f"http://{server_address}/queue", timeout=5)
except requests.exceptions.BaseException:
except requests.exceptions.RequestException:
print("Error while checking queue")
return None
queue_json = response.json()