Fix YAML Load error

mitre-attack · Feb 29, 2024 · 55b59f7 · 55b59f7
1 parent ea90218
commit 55b59f7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/scripts/generate_sensors.py b/scripts/generate_sensors.py
@@ -112,7 +112,10 @@ def generateSensorsForAnalytics(analytics, sensor_dict):
 data_models = []
 for dmf in data_model_files:
  print("working on {}".format(dmf))
- data_models.append(yaml.load(open(dmf).read()))
+ with open(dmf) as d:
+ data_files = yaml.safe_load(d)
+ data_models.append(data_files)
+ #data_models.append(yaml.load(open(dmf).read()))
 #data_models = [yaml.load(open(data_model_file).read()) for data_model_file in data_model_files]
 
 # Parse each analytic to find its data model references (if any)