API change in Ruamel >=0.15

cbg-ethz · Dec 26, 2023 · 9a3e502 · 9a3e502
1 parent e8897d4
commit 9a3e502
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/workflow/envs/timeline.yaml b/workflow/envs/timeline.yaml
@@ -6,5 +6,5 @@ dependencies:
   - python
   - numpy
   - pandas
-  - ruamel.yaml
+  - ruamel.yaml>=0.15
   - regex
diff --git a/workflow/scripts/file_parser.py b/workflow/scripts/file_parser.py
@@ -202,11 +202,8 @@ def main():
     datefmt = None
     if args.config_file:
         with open(args.config_file, "r") as stream:
-            try:
-                reg_data = ruamel.yaml.load(stream, Loader=ruamel.yaml.Loader)
-            except yaml.YAMLError as exc:
-                print(exc)
-                sys.exit(1)
+            yaml = ruamel.yaml.YAML(typ='rt')
+            reg_data = yaml.load(stream)
         if "sample" in reg_data:
             rxsam = regex.compile(reg_data.get("sample"))
         if "batch" in reg_data:
@@ -266,7 +263,8 @@ def main():
 
     if args.out_loc:
         with open(args.out_loc, "w") as outf:
-            ruamel.yaml.round_trip_dump(
+            yaml = ruamel.yaml.YAML()
+            yaml.dump(
                 {
                     "locations_list": list(
                         set(