Refactor yaml loading in scribereader.py

Yelp · Oct 21, 2024 · fb1e346 · fb1e346
1 parent e8e3c4d
commit fb1e346
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tron/utils/scribereader.py b/tron/utils/scribereader.py
@@ -87,7 +87,7 @@ def get_log_namespace(action_run_id: str, paasta_cluster: str) -> str:
     for ext in ["yaml", "yml"]:
         try:
             with open(f"/nail/etc/services/{namespace}/tron-{paasta_cluster}.{ext}") as f:
-                config = yaml.safe_load(f)
+                config = yaml.load(f, Loader=yaml.CSafeLoader)
                 service: Optional[str] = (
                     config.get(job_name, {}).get("actions", {}).get(action, {}).get("service", None)
                 )