From 0e04646cacda62bc474112071ecb58ec5df208fe Mon Sep 17 00:00:00 2001
From: Victor Chang <victortwchang@gmail.com>
Date: Tue, 21 Nov 2023 14:39:25 +0900
Subject: [PATCH] fix: deprecated safe_load

---
 simulation/main.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/simulation/main.py b/simulation/main.py
index e95cdde..7bcbfcc 100644
--- a/simulation/main.py
+++ b/simulation/main.py
@@ -2,7 +2,7 @@
 
 import sys
 import argparse
-import ruamel.yaml as yaml
+from ruamel.yaml import YAML
 from jobs import JobReader, RefJobReader
 from simulation import Simulation
 
@@ -10,7 +10,8 @@
 def get_configuration(config_file):
     with open(config_file, "r") as stream:
         try:
-            return yaml.safe_load(stream)
+            yaml = YAML(typ='safe', pure=True)
+            return yaml.load(stream)
         except yaml.YAMLError as e:
             print("Simulation config file {} is not in YAML format.".format(config_file))
             print(e)