diff --git a/capa/ida/ida_capa_explorer.py b/capa/ida/ida_capa_explorer.py index 801985bf..20757fc3 100644 --- a/capa/ida/ida_capa_explorer.py +++ b/capa/ida/ida_capa_explorer.py @@ -342,7 +342,9 @@ class CapaExplorerForm(idaapi.PluginForm): meta = capa.ida.helpers.collect_metadata() - capabilities, counts = capa.main.find_capabilities(rules, capa.features.extractors.ida.IdaFeatureExtractor(), True) + capabilities, counts = capa.main.find_capabilities( + rules, capa.features.extractors.ida.IdaFeatureExtractor(), True + ) meta.update(counts) # support binary files specifically for x86/AMD64 shellcode diff --git a/capa/main.py b/capa/main.py index 9cbb1589..b51e654d 100644 --- a/capa/main.py +++ b/capa/main.py @@ -96,12 +96,7 @@ def find_capabilities(ruleset, extractor, disable_progress=None): all_function_matches = collections.defaultdict(list) all_bb_matches = collections.defaultdict(list) - meta = { - "counts": { - "file": 0, - "functions": {}, - } - } + meta = {"counts": {"file": 0, "functions": {},}} for f in tqdm.tqdm(extractor.get_functions(), disable=disable_progress, unit=" functions"): function_matches, bb_matches, feature_count = find_function_capabilities(ruleset, extractor, f)