From dd2671aac23c7e8226a7adf24ef51d2aa8f70538 Mon Sep 17 00:00:00 2001 From: William Ballenthin Date: Mon, 14 Jun 2021 11:10:42 -0600 Subject: [PATCH] rules: fix types --- capa/rules.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/capa/rules.py b/capa/rules.py index 78c566fc..6fa9a4c4 100644 --- a/capa/rules.py +++ b/capa/rules.py @@ -145,7 +145,7 @@ class InvalidRuleSet(ValueError): return str(self) -def ensure_feature_valid_for_scope(scope: str, feature: Feature): +def ensure_feature_valid_for_scope(scope: str, feature: Union[Feature, Statement]): # if the given feature is a characteristic, # check that is a valid characteristic for the given scope. if (