From c4cad8e3018e26f697f4ab0c5926e0c93aa0315b Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 9 Nov 2022 15:21:52 +0100 Subject: [PATCH] Update `CLIPSegModelTester` (#20134) Co-authored-by: ydshieh --- tests/models/clipseg/test_modeling_clipseg.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/tests/models/clipseg/test_modeling_clipseg.py b/tests/models/clipseg/test_modeling_clipseg.py index 3a338ddbf8..8378e75a3f 100644 --- a/tests/models/clipseg/test_modeling_clipseg.py +++ b/tests/models/clipseg/test_modeling_clipseg.py @@ -345,10 +345,16 @@ class CLIPSegTextModelTest(ModelTesterMixin, unittest.TestCase): class CLIPSegModelTester: - def __init__(self, parent, is_training=True): + def __init__(self, parent, text_kwargs=None, vision_kwargs=None, is_training=True): + + if text_kwargs is None: + text_kwargs = {} + if vision_kwargs is None: + vision_kwargs = {} + self.parent = parent - self.text_model_tester = CLIPSegTextModelTester(parent) - self.vision_model_tester = CLIPSegVisionModelTester(parent) + self.text_model_tester = CLIPSegTextModelTester(parent, **text_kwargs) + self.vision_model_tester = CLIPSegVisionModelTester(parent, **vision_kwargs) self.is_training = is_training def prepare_config_and_inputs(self):