Test ignoring exceptions in cleanup of download queue

2024-08-30 20:32:17 +00:00 · 2024-02-29 00:04:08 -05:00
1 changed files with 9 additions and 3 deletions
--- a/invokeai/app/services/download/download_default.py
+++ b/invokeai/app/services/download/download_default.py
@ -228,10 +228,16 @@ class DownloadQueueService(DownloadQueueServiceBase):
            except (OSError, HTTPError) as excp:
                job.error_type = excp.__class__.__name__ + f"({str(excp)})"
                job.error = traceback.format_exc()
-                self._signal_job_error(job, excp)
+                try:
+                    self._signal_job_error(job, excp)
+                except:
+                    pass
            except DownloadJobCancelledException:
-                self._signal_job_cancelled(job)
-                self._cleanup_cancelled_job(job)
+                try:
+                    self._signal_job_cancelled(job)
+                    self._cleanup_cancelled_job(job)
+                except:
+                    pass

            finally:
                job.job_ended = get_iso_timestamp()