diff --git a/cbir/retrieval/retrieval.py b/cbir/retrieval/retrieval.py index 71acc1e..0cecace 100644 --- a/cbir/retrieval/retrieval.py +++ b/cbir/retrieval/retrieval.py @@ -124,6 +124,4 @@ def search( labels, distances = self.indexer.search(outputs, nrt_neigh) filenames = [self.store.get(str(l)) or "" for l in labels] - return [ - (filename, distance) for filename, distance in zip(filenames, distances) - ] + return list(zip(filenames, distances))