fixing test errors

Author: Lokiiiiii

test/unit/test_pytorch_xla.py

@@ -80,8 +80,8 @@ def is_oss_pt_xla_env():
 @pytest.mark.parametrize("instance_type", ["ml.p3.16xlarge", "ml.p3.2xlarge"])
 @pytest.mark.parametrize("cluster_size", [1, 4])
 class TestPyTorchXLARunner:
-    @patch.object(PyTorchXLARunner, "_check_compatibility")
     def test_setup(self, cluster, cluster_size, master, instance_type, num_gpus, *patches):
+        PyTorchXLARunner._check_compatibility = lambda x: None
         for rank, current_host in enumerate(cluster):
             print(f"Testing as host {rank+1}/{cluster_size}")
             runner = PyTorchXLARunner(