fix review comments

tools/test_runner.py

@@ -111,10 +111,7 @@ def model_parameter_names(model_file_name):
         model_proto = onnx.ModelProto()
         model_proto.ParseFromString(data_str)
         init_names = set([(i.name) for i in model_proto.graph.initializer])
-        param_names = []
-        for input in model_proto.graph.input:
-            if input.name not in init_names:
-                param_names.append(input.name)
+        param_names = [input.name for input in model_proto.graph.input if input.name not in init_names]
 
         return param_names
 
@@ -149,7 +146,7 @@ def get_input_shapes(sample_case, param_names):
         return param_shape_map
 
     for name in param_names:
-        if not name in param_shape_map.keys():
+        if not name in param_shape_map:
             print("Input {} does not exist!".format(name))
             sys.exit()
 
@@ -160,7 +157,6 @@ def run_one_case(model, param_map):
     # convert np array to model argument
     pp = {}
     for key, val in param_map.items():
-        #print("input: {} = {}".format(key, val))
         pp[key] = migraphx.argument(val)
 
     # run the model
@@ -257,9 +253,6 @@ def main():
 
         # run the model and return outputs
         output_data = run_one_case(model, input_data)
-        # gold_output_data = []
-        # for i in range(len((output_data))):
-        #     gold_output_data.append(gold_outputs[output_names[i]])
 
         # check output correctness
         ret = check_correctness(gold_outputs, output_data)