From e6bdb00d317aea3ca7f85b097c120b2db39314c2 Mon Sep 17 00:00:00 2001 From: Yash Katariya Date: Fri, 16 Sep 2022 15:25:43 -0700 Subject: [PATCH] Skip remote_transfer_test because Array does not have the xla_shape method since its deprecated. PiperOrigin-RevId: 474913967 --- tests/remote_transfer_test.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/remote_transfer_test.py b/tests/remote_transfer_test.py index fe00f78c2..8bb9e6ef1 100644 --- a/tests/remote_transfer_test.py +++ b/tests/remote_transfer_test.py @@ -31,7 +31,10 @@ class RemoteTransferTest(jtu.JaxTestCase): @jtu.skip_on_devices("gpu") def test_remote_transfer(self): if jax.device_count() < 2: - raise unittest.SkipTest("Remote transfer requires at lest 2 devices") + raise unittest.SkipTest("Remote transfer requires at least 2 devices") + if config.jax_array: + raise unittest.SkipTest("Array does not have xla_shape method since " + "it is deprecated.") dev_a, dev_b = jax.local_devices()[:2] if "libtpu" in jax.local_devices()[0].client.platform_version: raise unittest.SkipTest("Test does not yet work on cloud TPU")