remove mistakenly added assert

Author: hshahTT

test/spmd/test_xla_sharding.py

@@ -1189,10 +1189,6 @@ def test_mark_shard_scalar(self):
       self.assertIsInstance(shard.indices, type(Ellipsis))
       self.assertEqual(shard.replica_id, i)
 
-    # It looks like mesh_shape attribute is never implemented.
-    with self.assertRaises(AttributeError):
-      xt.mesh_shape
-
   def test_global_mesh(self):
     expected_mesh = self._get_mesh((1, self.n_devices))
     xs.set_global_mesh(expected_mesh)

torch_xla/distributed/spmd/debugging.py

@@ -2,7 +2,7 @@
 import functools
 import string
 import sys
-from typing import Any, Callable, Optional, Union, Tuple
+from typing import Any, Callable, Optional, Union
 import weakref
 
 import numpy as np