fix tests chen only using tf

cf9c1cbb · thomwolf · 2167e366 · cf9c1cbb
Commit cf9c1cbb authored Sep 24, 2019 by thomwolf
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

pytorch_transformers/tests/modeling_tf_common_test.py pytorch_transformers/tests/modeling_tf_common_test.py +2 -1

No files found.
--- a/pytorch_transformers/tests/modeling_tf_common_test.py
+++ b/pytorch_transformers/tests/modeling_tf_common_test.py
@@ -69,7 +69,8 @@ class TFCommonTestCases:

        def test_pt_tf_model_equivalence(self):
            if not is_torch_available():
-                pass
+                return
+
            import pytorch_transformers

            config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()