Updated PyTorch Eye Op tests

twaclaw · Diego Sandoval · commit f5d818f6d40b · 2024-07-04T18:50:04.000Z
diff --git a/tests/link/pytorch/test_basic.py b/tests/link/pytorch/test_basic.py
@@ -290,5 +290,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
`@@ -290,5 +290,5 @@ def test_eye():`
`290`	`290`	`for _k in list(range(_M + 2)) + [-x for x in range(1, _N + 2)]:`
`291`	`291`	`compare_pytorch_and_py(`
`292`	`292`	`FunctionGraph([N, M, k], [out]),`
`293`		`- [np.array(_N + 1), np.array(_M + 1), np.array(_k)],`
	`293`	`+ [np.array(_N), np.array(_M), np.array(_k)],`
`294`	`294`	`)`