From 4e43ff9086af7c784f779a8ae3109d5f8fa95ebb Mon Sep 17 00:00:00 2001
From: k4yt3x <k4yt3x@k4yt3x.com>
Date: Mon, 11 May 2020 21:01:09 -0400
Subject: [PATCH] fixed Anime4KCPP attribute issue

---
 src/upscaler.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/upscaler.py b/src/upscaler.py
index f3b55a7..dbbfad9 100755
--- a/src/upscaler.py
+++ b/src/upscaler.py
@@ -485,6 +485,9 @@ class Upscaler:
                         self.process_pool.append(self.driver_object.upscale(self.current_input_file, output_path))
                         self._wait()
                         Avalon.info(_('Upscaling completed'))
+                        self.processing_queue.task_done()
+                        self.total_processed += 1
+                        continue
 
                     else:
                         self.create_temp_directories()
@@ -606,7 +609,7 @@ class Upscaler:
                 self.total_processed += 1
 
         except (Exception, KeyboardInterrupt, SystemExit) as e:
-            with contextlib.suppress(ValueError):
+            with contextlib.suppress(ValueError, AttributeError):
                 self.cleanup_temp_directories()
                 self.running = False
             raise e