Fix tests for engine

titu1994 · Oct 12, 2018 · 51e2da5 · 51e2da5
1 parent 3d58feb
commit 51e2da5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/core/test_engine.py b/tests/core/test_engine.py
@@ -12,6 +12,12 @@
 
 warnings.simplefilter('ignore', DeprecationWarning)
 
+# compatible with both Python 2 and 3
+try:
+    FileNotFoundError
+except NameError:
+    FileNotFoundError = IOError
+
 
 # wrapper function to clean up saved files and be deterministic
 def optimizer_wrapper(func):