Skip to content

Commit d81e7ba

Browse files
committed
misc: Tweak memory estimation parsing
1 parent e064d2f commit d81e7ba

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

tests/test_operator.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2073,7 +2073,8 @@ def parse_output(self, output, expected):
20732073
# Check that no allocation occurs as estimate_memory should avoid data touch
20742074
assert "Allocating" not in output.text
20752075

2076-
name, host, device = output.records[-1].message.split()
2076+
parsed = output.records[-1].message.split()
2077+
name, host, device = parsed[:3]
20772078
extracted = (name, int(host), int(device))
20782079

20792080
assert extracted == expected

0 commit comments

Comments
 (0)