Fix/tests in master (#2592)

mandiant · Feb 4, 2025 · 40203a0 · 40203a0
1 parent 5467fac
commit 40203a0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/capa/render/utils.py b/capa/render/utils.py
@@ -58,6 +58,8 @@ def capability_rules(doc: rd.ResultDocument) -> Iterator[rd.RuleMatches]:
     for _, _, rule in sort_rules(doc.rules):
         if rule.meta.lib:
             continue
+        if (rule.meta.namespace or "").startswith("internal/"):
+            continue
         if rule.meta.is_subscope_rule:
             continue
         if rule.meta.maec.analysis_conclusion:

diff --git a/capa/rules/__init__.py b/capa/rules/__init__.py
@@ -2227,8 +2227,8 @@ def get_rules(
         on_load_rule(path, i, total_rule_count)
 
         try:
-            rule = capa.rules.Rule.from_yaml(content.decode("utf-8"))
-        except capa.rules.InvalidRule:
+            rule = Rule.from_yaml(content.decode("utf-8"))
+        except InvalidRule:
             raise
         else:
             rule.meta["capa/path"] = path.as_posix()
@@ -2237,7 +2237,7 @@ def get_rules(
             rules.append(rule)
             logger.debug("loaded rule: '%s' with scope: %s", rule.name, rule.scopes)
 
-    ruleset = capa.rules.RuleSet(rules)
+    ruleset = RuleSet(rules)
 
     capa.rules.cache.cache_ruleset(cache_dir, ruleset)