Added Python 2 and 3 compatible temp directory utility function

scidash · Apr 18, 2019 · 2b046a4 · 2b046a4
1 parent 619b6c1
commit 2b046a4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -10,3 +10,4 @@ ipykernel
 nbformat
 gitpython
 cerberus>=1.2
+backports.tempfile
diff --git a/sciunit/utils.py b/sciunit/utils.py
@@ -14,6 +14,10 @@
 import contextlib
 from io import TextIOWrapper, StringIO
 from datetime import datetime
+try:
+    from tempfile import TemporaryDirectory
+except ImportError:
+    from backports.tempfile import TemporaryDirectory
 
 import bs4
 import nbformat