From 282c0c26559dcdc4e053a9b68289f81fb247472d Mon Sep 17 00:00:00 2001 From: William Ballenthin Date: Fri, 27 Aug 2021 13:00:40 -0600 Subject: [PATCH] lint: guide mypy typing to address CI issues --- scripts/lint.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/scripts/lint.py b/scripts/lint.py index ce675f41..f2f6f6e6 100644 --- a/scripts/lint.py +++ b/scripts/lint.py @@ -234,8 +234,13 @@ def get_sample_capabilities(ctx: Context, path: Path) -> Set[str]: extractor = capa.main.get_extractor( nice_path, "auto", capa.main.BACKEND_VIV, DEFAULT_SIGNATURES, False, disable_progress=True ) + capabilities, _ = capa.main.find_capabilities(ctx.rules, extractor, disable_progress=True) - capabilities = set(capabilities.keys()) + # mypy doesn't seem to be happy with the MatchResults type alias & set(...keys())? + # so we ignore a few types here. + capabilities = set(capabilities.keys()) # type: ignore + assert isinstance(capabilities, set) + logger.debug("computed results: %s: %d capabilities", nice_path, len(capabilities)) ctx.capabilities_by_sample[path] = capabilities