@@ -228,7 +228,7 @@ def test_graph_conditional_if(init_cuda, condition_value):
228228 try :
229229 handle = gb .create_conditional_handle ()
230230 except RuntimeError as e :
231- with pytest .raises (RuntimeError , match = "^Driver version \d+ does not support conditional handles" ):
231+ with pytest .raises (RuntimeError , match = r "^Driver version \d+ does not support conditional handles" ):
232232 raise e
233233 gb .end_building ()
234234 pytest .skip ("Driver does not support conditional handle" )
@@ -289,7 +289,7 @@ def test_graph_conditional_if_else(init_cuda, condition_value):
289289 try :
290290 gb_if , gb_else = gb .if_else (handle )
291291 except RuntimeError as e :
292- with pytest .raises (RuntimeError , match = "^Driver version \d+ does not support conditional if-else" ):
292+ with pytest .raises (RuntimeError , match = r "^Driver version \d+ does not support conditional if-else" ):
293293 raise e
294294 gb .end_building ()
295295 pytest .skip ("Driver does not support conditional if-else" )
@@ -357,7 +357,7 @@ def test_graph_conditional_switch(init_cuda, condition_value):
357357 try :
358358 gb_case = list (gb .switch (handle , 3 ))
359359 except RuntimeError as e :
360- with pytest .raises (RuntimeError , match = "^Driver version \d+ does not support conditional switch" ):
360+ with pytest .raises (RuntimeError , match = r "^Driver version \d+ does not support conditional switch" ):
361361 raise e
362362 gb .end_building ()
363363 pytest .skip ("Driver does not support conditional switch" )
@@ -534,7 +534,7 @@ def build_graph(condition_value):
534534 try :
535535 gb_case = list (gb .switch (handle , 3 ))
536536 except RuntimeError as e :
537- with pytest .raises (RuntimeError , match = "^Driver version \d+ does not support conditional switch" ):
537+ with pytest .raises (RuntimeError , match = r "^Driver version \d+ does not support conditional switch" ):
538538 raise e
539539 gb .end_building ()
540540 pytest .skip ("Driver does not support conditional switch" )
0 commit comments