Model parallel tests should return, not pass in non model parallel settings. (#8825)

2020-11-27 16:41:29 -05:00
parent edbff1fd00
commit 18c32eeb21
1 changed files with 2 additions and 2 deletions
--- 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()