pyright errors

Author: NeilGirdhar

tests/test_funcs.py

@@ -465,7 +465,7 @@ def test_dims_and_classes(self, xp: ModuleType, n_dim: int, num_classes: int):
         assert y.shape == (*x.shape, num_classes)
         for *i_list, j in ndindex(*shape, num_classes):
             i = tuple(i_list)
-            assert float(y[*i, j]) == (int(x[i]) == j)
+            assert float(y[(*i, j)]) == (int(x[i]) == j)
 
     def test_basic(self, xp: ModuleType):
         actual = one_hot(xp.asarray([0, 1, 2]), 3)
@@ -516,7 +516,7 @@ def test_axis(self, xp: ModuleType):
 
     def test_non_integer(self, xp: ModuleType):
         with pytest.raises(TypeError):
-            one_hot(xp.asarray([1.0]), 3)
+            _ = one_hot(xp.asarray([1.0]), 3)
 
 
 @pytest.mark.skip_xp_backend(