From 74eb14f584c5fb61e22e60c2b8d8159ba1494d0b Mon Sep 17 00:00:00 2001
From: JanLJL <jan.laukemann@fau.de>
Date: Tue, 3 Sep 2024 14:22:19 +0200
Subject: [PATCH] fixed YAML output

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

diff --git a/osaca/osaca.py b/osaca/osaca.py
index 8a2fbf0..d63af92 100644
--- a/osaca/osaca.py
+++ b/osaca/osaca.py
@@ -7,7 +7,7 @@ import re
 import sys
 from functools import lru_cache
 
-import ruamel.yaml
+from ruamel.yaml import YAML
 
 from osaca.db_interface import import_benchmark_output, sanity_check
 from osaca.frontend import Frontend
@@ -375,7 +375,8 @@ def inspect(args, output_file=sys.stdout):
         file=output_file,
     )
     if args.yaml_out is not None:
-        ruamel.yaml.dump(
+        yaml = YAML(typ='unsafe', pure=True)
+        yaml.dump(
             frontend.full_analysis_dict(
                 kernel,
                 kernel_graph,