Fix YAML loading of analytics

mitre-attack · Feb 29, 2024 · fa64d0f · fa64d0f
1 parent 85c9292
commit fa64d0f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/scripts/generate_analytics.py b/scripts/generate_analytics.py
@@ -22,7 +22,9 @@
 analytics = []
 for af in analytics_files:
  print("appending {}".format(af))
- analytics.append(yaml.load(open(af,encoding='utf-8').read()))
+ with open(af, encoding='utf-8') as f:
+ analytic_data = yaml.safe_load(f)
+ analytics.append(analytic_data)
 #analytics = [yaml.load(open(analytic_file).read()) for analytic_file in analytics_files]
 
 # Load ATT&CK content, which is needed to get names for technique IDs