From 0f10a9ce636566b86a080ee9dab8e48aa016e8f4 Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Sat, 26 Aug 2023 17:19:35 +0200
Subject: [PATCH] catch numpy warning

---
 modules/processing.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/modules/processing.py b/modules/processing.py
index e73b4efec..d561a6c1d 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -583,14 +583,15 @@ def validate_sample(sample):
     try:
         sample = sample.astype(np.uint8)
         return sample
-    except Exception as e:
+    except (Warning, Exception) as e:
         shared.log.error(f'Failed to validate sample values: {e}')
         ok = False
     if not ok:
         try:
             sample = np.nan_to_num(sample, nan=0, posinf=255, neginf=0)
             sample = sample.astype(np.uint8)
-        except Exception as e:
+            shared.log.debug('Corrected sample values')
+        except (Warning, Exception) as e:
             shared.log.error(f'Failed to correct sample values: {e}')
             sample = np.zeros_like(sample)
             sample = sample.astype(np.uint8)