Update default.py

capa/render/default.py

@@ -8,6 +8,7 @@
 
 import pickle
 import collections
+from pathlib import Path
 
 import tabulate
 
@@ -85,8 +86,13 @@ def render_capabilities(doc: rd.ResultDocument, ostream: StringIO):
     """
 
     def load_rules_prevalence(filename: str) -> dict:
-        with open(filename, "rb") as pickle_file:
-            return pickle.load(pickle_file)
+        try:
+            with Path(filename).open("rb") as pickle_file:
+                return pickle.load(pickle_file)
+        except FileNotFoundError:
+            raise FileNotFoundError(f"File '{filename}' not found.")
+        except Exception as e:
+            raise RuntimeError(f"An error occurred while loading '{filename}': {e}")
 
     subrule_matches = find_subrule_matches(doc)
     rules_prevalence = load_rules_prevalence("./assets/rules_prevalence.pickle")
@@ -101,7 +107,7 @@ def load_rules_prevalence(filename: str) -> dict:
         count = len(rule.matches)
         matches = f"({count} matches)" if count > 1 else ""
 
-        rule_prevalence = float(rules_prevalence.get(rule.meta.name + "\n", 0))
+        rule_prevalence = float(rules_prevalence.get(rule.meta.name + "\n", 0) / 593)
         if rule_prevalence < 0:
             raise ValueError("Match probability cannot be negative")