From 18c32eeb21afa1d883094af54ba6321aa2d4c8db Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Fri, 27 Nov 2020 16:41:29 -0500 Subject: [PATCH] Model parallel tests should return, not pass in non model parallel settings. (#8825) --- tests/test_modeling_common.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index 977a4a1e8b..9375e11085 100755 --- a/tests/test_modeling_common.py +++ b/tests/test_modeling_common.py @@ -1022,7 +1022,7 @@ class ModelTesterMixin: @require_torch_multi_gpu def test_model_parallelization(self): if not self.test_model_parallel: - pass + return import subprocess @@ -1079,7 +1079,7 @@ class ModelTesterMixin: @require_torch_multi_gpu def test_model_parallel_equal_results(self): if not self.test_model_parallel: - pass + return config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()