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

18c32eeb · Lysandre Debut · GitHub · edbff1fd · 18c32eeb
Unverified Commit 18c32eeb authored Nov 27, 2020 by Lysandre Debut Committed by GitHub Nov 27, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tests/test_modeling_common.py tests/test_modeling_common.py +2 -2

No files found.
--- 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()