@@ -63,13 +63,13 @@ def test_run_tf_model(self):
6363
6464 con = self .get_client ()
6565 con .modelset ('m' , Backend .tf , Device .cpu , model_pb ,
66- input = ['a' , 'b' ], output = 'mul' )
66+ inputs = ['a' , 'b' ], outputs = 'mul' )
6767
6868 # wrong model
6969 self .assertRaises (ResponseError ,
7070 con .modelset , 'm' , Backend .tf , Device .cpu ,
7171 wrong_model_pb ,
72- input = ['a' , 'b' ], output = 'mul' )
72+ inputs = ['a' , 'b' ], outputs = 'mul' )
7373 # missing inputs/outputs
7474 self .assertRaises (ValueError ,
7575 con .modelset , 'm' , Backend .tf , Device .cpu ,
@@ -79,7 +79,7 @@ def test_run_tf_model(self):
7979 self .assertRaises (ResponseError ,
8080 con .modelset , 'm' , Backend .torch , Device .cpu ,
8181 model_pb ,
82- input = ['a' , 'b' ], output = 'mul' )
82+ inputs = ['a' , 'b' ], outputs = 'mul' )
8383
8484 con .tensorset ('a' , Tensor .scalar (DType .float , 2 , 3 ))
8585 con .tensorset ('b' , Tensor .scalar (DType .float , 2 , 3 ))
@@ -99,8 +99,8 @@ def bar(a, b):
9999 con .tensorset ('b' , Tensor .scalar (DType .float , 2 , 3 ))
100100 # try with bad arguments:
101101 self .assertRaises (ResponseError ,
102- con .scriptrun , 'ket' , 'bar' , input = 'a' , output = 'c' )
103- con .scriptrun ('ket' , 'bar' , input = ['a' , 'b' ], output = 'c' )
102+ con .scriptrun , 'ket' , 'bar' , inputs = 'a' , outputs = 'c' )
103+ con .scriptrun ('ket' , 'bar' , inputs = ['a' , 'b' ], outputs = 'c' )
104104 tensor = con .tensorget ('c' , as_type = Tensor )
105105 self .assertEqual ([4 , 6 ], tensor .value )
106106
0 commit comments