From fb2813de52865950b67d60f542c70e96a60e176b Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Tue, 24 Jun 2025 14:42:08 +0530 Subject: [PATCH] skip instead of returning. --- tests/models/test_modeling_common.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/models/test_modeling_common.py b/tests/models/test_modeling_common.py index e8b41ddbfd87..f9a150181410 100644 --- a/tests/models/test_modeling_common.py +++ b/tests/models/test_modeling_common.py @@ -2108,7 +2108,7 @@ def test_hotswapping_compiled_model_linear(self, rank0, rank1): @parameterized.expand([(11, 11), (7, 13), (13, 7)]) # important to test small to large and vice versa def test_hotswapping_compiled_model_conv2d(self, rank0, rank1): if "unet" not in self.model_class.__name__.lower(): - return + pytest.skip("Test only applies to UNet.") # It's important to add this context to raise an error on recompilation target_modules = ["conv", "conv1", "conv2"] @@ -2118,7 +2118,7 @@ def test_hotswapping_compiled_model_conv2d(self, rank0, rank1): @parameterized.expand([(11, 11), (7, 13), (13, 7)]) # important to test small to large and vice versa def test_hotswapping_compiled_model_both_linear_and_conv2d(self, rank0, rank1): if "unet" not in self.model_class.__name__.lower(): - return + pytest.skip("Test only applies to UNet.") # It's important to add this context to raise an error on recompilation target_modules = ["to_q", "conv"]