Skip to content

Commit 7eba8ca

Browse files
committed
Fixed Edge Case for Diff Given < 3.7
1 parent 552a00e commit 7eba8ca

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

src/inline/plugin.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -603,15 +603,22 @@ def parse_diff_given(self, node):
603603
PROPERTY = 0
604604
VALUES = 1
605605

606+
if sys.version_info >= (3, 8, 0):
607+
attr_name = "value"
608+
else:
609+
attr_name = "s"
610+
611+
606612
if len(node.args) == 2:
607613
if self.cur_inline_test.parameterized:
608614
raise MalformedException("inline test: Parameterized inline tests currently do not support differential tests.")
609615
else:
610616
devices = []
611617
for elt in node.args[VALUES].elts:
612-
if elt.value not in {"cpu", "cuda", "mps"}:
618+
value = getattr(elt, attr_name)
619+
if value not in {"cpu", "cuda", "mps"}:
613620
raise MalformedException(f"Invalid device: {elt.value}. Must be one of ['cpu', 'cuda', 'mps']")
614-
devices.append(elt.value)
621+
devices.append(value)
615622
setattr(self.cur_inline_test, node.args[PROPERTY].id, devices)
616623
else:
617624
raise MalformedException("inline test: invalid diff_given(), expected 2 args")

0 commit comments

Comments
 (0)