Commit 1bbf8bdc authored by rusty1s's avatar rusty1s
Browse files

new index [ci skip]

parent b8166f31
...@@ -17,6 +17,6 @@ def test_rw(device): ...@@ -17,6 +17,6 @@ def test_rw(device):
for n in range(start.size(0)): for n in range(start.size(0)):
cur = start[n].item() cur = start[n].item()
for le in range(1, walk_length): for i in range(1, walk_length):
assert out[n, le].item() in col[row == cur].tolist() assert out[n, i].item() in col[row == cur].tolist()
cur = out[n, le].item() cur = out[n, i].item()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment