[models] fix bug in model lookup with multiple formats

mlonmcu/flow/tvm/backend/model_info.py

@@ -144,7 +144,7 @@ def __init__(self, mod_text, fix_names=False):
         super().__init__(in_tensors, out_tensors)
 
 
-def get_tfgraph_inout(graph, graph_def):
+def get_tfgraph_inout(graph):
     ops = graph.get_operations()
     outputs_set = set(ops)
     inputs = []
@@ -166,7 +166,7 @@ def __init__(self, model_file):
             graph_def.ParseFromString(f.read())
         with tf.Graph().as_default() as graph:
             tf.import_graph_def(graph_def)
-            inputs, outputs = get_tfgraph_inout(graph, graph_def)
+            inputs, outputs = get_tfgraph_inout(graph)
         in_tensors = [TensorInfo(t.name, t.shape.as_list(), t.dtype.name) for op in inputs for t in op.outputs]
         out_tensors = [TensorInfo(t.name, t.shape.as_list(), t.dtype.name) for op in outputs for t in op.outputs]
         super().__init__(in_tensors, out_tensors)

mlonmcu/models/lookup.py

@@ -59,6 +59,7 @@ def list_models(directory, depth=1, formats=None, config=None):  # TODO: get con
     config = config if config is not None else {}
     formats = formats if formats else [ModelFormats.TFLITE]
     assert len(formats) > 0, "No formats provided for model lookup"
+    models = []
     for fmt in formats:
         if depth != 1:
             raise NotImplementedError  # TODO: implement for arm ml zoo
@@ -68,7 +69,6 @@ def list_models(directory, depth=1, formats=None, config=None):  # TODO: get con
             logger.debug("Not a directory: %s", str(directory))
             return []
         subdirs = [Path(directory) / o for o in os.listdir(directory) if os.path.isdir(os.path.join(directory, o))]
-        models = []
         for subdir in subdirs:
             dirname = subdir.name
             if dirname.startswith("."):