Fix TensorFlow CI failures

Author: njzjz

src/array_api_compat/tensorflow/__init__.py

@@ -78,7 +78,7 @@ def bool_getitem(self, key):
         x_shape = py_tuple(self.shape)
         key_shape = py_tuple(key.shape)
         if key.ndim > self.ndim or py_any(
-            ks not in (xs, 0) for xs, ks in zip(x_shape, key_shape)
+            ks not in (xs, 0) for xs, ks in zip(x_shape, key_shape, strict=False)
         ):
             raise IndexError("boolean index did not match indexed array")
         if key.ndim == 0:
@@ -100,13 +100,13 @@ def high_dim_int_getitem(self, key):
         if not py_all(isinstance(k, py_int) and not isinstance(k, py_bool) for k in key):
             return None
         normalized = []
-        for k, side in zip(key, shape):
+        for k, side in zip(key, shape, strict=True):
             k = k + side if k < 0 else k
             if k < 0 or k >= side:
                 raise IndexError("index out of bounds")
             normalized.append(k)
         flat_index = 0
-        for k, stride in zip(normalized, _strides(shape)):
+        for k, stride in zip(normalized, _strides(shape), strict=True):
             flat_index += k * stride
         return _tf.gather(_tf.reshape(self, (-1,)), flat_index)
 

src/array_api_compat/tensorflow/_aliases.py

@@ -666,9 +666,9 @@ def full(
     del kwargs
     if isinstance(shape, int):
         shape = (shape,)
-    value = tf.convert_to_tensor(fill_value, dtype=dtype)
     with tf.device(device):
-        return tf.fill(shape, value)
+        value = _to_tensor(fill_value, dtype=dtype)
+        return tf.broadcast_to(value, shape)
 
 
 def full_like(