Support Python 2 for ImportErrors

mrocklin · Aug 1, 2018 · be1be64 · be1be64
1 parent 712ba26
commit be1be64
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/dask/array/__init__.py b/dask/array/__init__.py
@@ -63,5 +63,4 @@
            "Please either conda or pip install as follows:\n\n"
            "  conda install dask                 # either conda install\n"
            "  pip install dask[array] --upgrade  # or pip install")
-    e.msg += "\n\n" + msg
-    raise e
+    raise ImportError(str(e) + '\n\n' + msg)
diff --git a/dask/bag/__init__.py b/dask/bag/__init__.py
@@ -13,5 +13,4 @@
            "Please either conda or pip install as follows:\n\n"
            "  conda install dask               # either conda install\n"
            "  pip install dask[bag] --upgrade  # or pip install")
-    e.msg += "\n\n" + msg
-    raise e
+    raise ImportError(str(e) + '\n\n' + msg)
diff --git a/dask/dataframe/__init__.py b/dask/dataframe/__init__.py
@@ -28,5 +28,4 @@
            "Please either conda or pip install as follows:\n\n"
            "  conda install dask                     # either conda install\n"
            "  pip install dask[dataframe] --upgrade  # or pip install")
-    e.msg += "\n\n" + msg
-    raise e
+    raise ImportError(str(e) + '\n\n' + msg)