File tree Expand file tree Collapse file tree 1 file changed +9
-2
lines changed
Expand file tree Collapse file tree 1 file changed +9
-2
lines changed Original file line number Diff line number Diff 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" )
You can’t perform that action at this time.
0 commit comments