@@ -63,13 +63,13 @@ def test_run_tf_model(self):
63
63
64
64
con = self .get_client ()
65
65
con .modelset ('m' , Backend .tf , Device .cpu , model_pb ,
66
- input = ['a' , 'b' ], output = 'mul' )
66
+ inputs = ['a' , 'b' ], outputs = 'mul' )
67
67
68
68
# wrong model
69
69
self .assertRaises (ResponseError ,
70
70
con .modelset , 'm' , Backend .tf , Device .cpu ,
71
71
wrong_model_pb ,
72
- input = ['a' , 'b' ], output = 'mul' )
72
+ inputs = ['a' , 'b' ], outputs = 'mul' )
73
73
# missing inputs/outputs
74
74
self .assertRaises (ValueError ,
75
75
con .modelset , 'm' , Backend .tf , Device .cpu ,
@@ -79,7 +79,7 @@ def test_run_tf_model(self):
79
79
self .assertRaises (ResponseError ,
80
80
con .modelset , 'm' , Backend .torch , Device .cpu ,
81
81
model_pb ,
82
- input = ['a' , 'b' ], output = 'mul' )
82
+ inputs = ['a' , 'b' ], outputs = 'mul' )
83
83
84
84
con .tensorset ('a' , Tensor .scalar (DType .float , 2 , 3 ))
85
85
con .tensorset ('b' , Tensor .scalar (DType .float , 2 , 3 ))
@@ -99,8 +99,8 @@ def bar(a, b):
99
99
con .tensorset ('b' , Tensor .scalar (DType .float , 2 , 3 ))
100
100
# try with bad arguments:
101
101
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' )
104
104
tensor = con .tensorget ('c' , as_type = Tensor )
105
105
self .assertEqual ([4 , 6 ], tensor .value )
106
106
0 commit comments