add conditional import to support conda package manager

khanlab · Nov 22, 2024 · 64fc92e · 64fc92e
1 parent 0f07ece
commit 64fc92e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/snakebids/io/yaml.py b/snakebids/io/yaml.py
@@ -2,7 +2,10 @@
 from pathlib import Path, PosixPath, WindowsPath
 from typing import Any, OrderedDict
 
-from ruamel.yaml import YAML, Dumper
+try:
+    from ruamel.yaml import YAML, Dumper
+except ImportError:
+    from ruamel_yaml import YAML, Dumper
 
 
 def get_yaml_io():