Updated PyTorch Eye Op tests

twaclaw · twaclaw · commit f45b27792e0e · 2024-07-02T20:06:44.000Z
diff --git a/tests/link/pytorch/test_basic.py b/tests/link/pytorch/test_basic.py
@@ -250,5 +250,5 @@ def test_eye():
             for _k in list(range(_M + 2)) + [-x for x in range(1, _N + 2)]:
                 compare_pytorch_and_py(
                     FunctionGraph([N, M, k], [out]),
-                    [np.array(_N + 1), np.array(_M + 1), np.array(_k)],
+                    [np.array(_N), np.array(_M), np.array(_k)],
                 )

Original file line number	Diff line number	Diff line change
`@@ -250,5 +250,5 @@ def test_eye():`
`250`	`250`	`for _k in list(range(_M + 2)) + [-x for x in range(1, _N + 2)]:`
`251`	`251`	`compare_pytorch_and_py(`
`252`	`252`	`FunctionGraph([N, M, k], [out]),`
`253`		`- [np.array(_N + 1), np.array(_M + 1), np.array(_k)],`
	`253`	`+ [np.array(_N), np.array(_M), np.array(_k)],`
`254`	`254`	`)`