From 87c3cf3e4f70d81a78335d3b3979a05cd797d6f0 Mon Sep 17 00:00:00 2001 From: Bingsu Date: Thu, 22 Jun 2023 11:23:58 +0900 Subject: [PATCH] style: ruff RET --- adetailer/mask.py | 4 +--- scripts/!adetailer.py | 3 +-- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/adetailer/mask.py b/adetailer/mask.py index 48f9273..9209b45 100644 --- a/adetailer/mask.py +++ b/adetailer/mask.py @@ -129,9 +129,7 @@ def mask_preprocess( masks = [dilate_erode(m, kernel) for m in masks] masks = [m for m in masks if not is_all_black(m)] - masks = mask_merge_invert(masks, mode=merge_invert) - - return masks + return mask_merge_invert(masks, mode=merge_invert) # Bbox sorting diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py index d7b9058..6e00e5b 100644 --- a/scripts/!adetailer.py +++ b/scripts/!adetailer.py @@ -423,8 +423,7 @@ class AfterDetailerScript(scripts.Script): def sort_bboxes(self, pred: PredictOutput) -> PredictOutput: sortby = opts.data.get("ad_bbox_sortby", BBOX_SORTBY[0]) sortby_idx = BBOX_SORTBY.index(sortby) - pred = sort_bboxes(pred, sortby_idx) - return pred + return sort_bboxes(pred, sortby_idx) def pred_preprocessing(self, pred: PredictOutput, args: ADetailerArgs): pred = filter_by_ratio(