From 20c37d41d0b688732679d34d4d93a17879423f3c Mon Sep 17 00:00:00 2001 From: naelsondouglas Date: Sun, 17 Oct 2021 15:49:16 -0300 Subject: [PATCH] Removed a literal comparison pitfall from the code --- photo_wct.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/photo_wct.py b/photo_wct.py index e274b18..ccc0488 100644 --- a/photo_wct.py +++ b/photo_wct.py @@ -54,7 +54,7 @@ def transform(self, cont_img, styl_img, cont_seg, styl_seg): return Im1 def __compute_label_info(self, cont_seg, styl_seg): - if cont_seg.size == False or styl_seg.size == False: + if cont_seg.size is False or styl_seg.size is False: return max_label = np.max(cont_seg) + 1 self.label_set = np.unique(cont_seg)