#57 - hackish way to mage ruamel work on github

zerwes · Oct 19, 2024 · d51f082 · d51f082
1 parent 16db5f5
commit d51f082
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/test/test_ruamel_yaml_loader.py b/test/test_ruamel_yaml_loader.py
@@ -4,7 +4,10 @@
 import os
 import logging
 import platform
-from ruamel.yaml import YAML
+try:
+    from ruamel_yaml import YAML
+except ModuleNotFoundError:
+    from ruamel.yaml import YAML
 
 basepath = os.path.dirname(os.path.realpath(__file__))
 sys.path.insert(0, os.path.dirname(basepath))