Catch exceptions from yaml

fermiPy · Jul 9, 2019 · af36e24 · af36e24
1 parent 15a0522
commit af36e24
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dmsky/utils/tools.py b/dmsky/utils/tools.py
@@ -14,7 +14,7 @@ def yaml_load(filename):
         raise Exception('File does not exist: %s' % (filename))
     try:
         ret = yaml.load(open(filename), Loader=yaml.CLoader)
-    except IOError:
+    except (IOError, AttributeError):
         ret = yaml.load(open(filename), Loader=yaml.Loader)
     return ret
 
@@ -33,7 +33,7 @@ def yaml_dump(x, filename):
 
     try:
         out.write(yaml.dump(x, Dumper=yaml.CDumper))
-    except IOError:
+    except (IOError, AttributeError):
         out.write(yaml.dump(x, Dumper=yaml.Dumper))
     out.close()