catch correct error

cooperative-computing-lab · Feb 10, 2025 · 0068f4e · 0068f4e
1 parent e91da4a
commit 0068f4e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/taskvine/src/bindings/python3/ndcctools/taskvine/__init__.py b/taskvine/src/bindings/python3/ndcctools/taskvine/__init__.py
@@ -75,7 +75,7 @@ class DaskVineWarning(UserWarning):
 
     from .dask_executor import DaskVine
     from .dask_dag import DaskVineDag
-except ImportError:
+except (ImportError, ModuleNotFoundError):
     warnings.warn("Dask >= 2024.12.0 not available, using DaskVine legacy task graph representation.", DaskVineWarning)
 
     from .compat import DaskVine

diff --git a/taskvine/src/bindings/python3/ndcctools/taskvine/compat/__init__.py b/taskvine/src/bindings/python3/ndcctools/taskvine/compat/__init__.py
@@ -35,7 +35,7 @@
 
     if vd >= vr:
         warnings.warn("ndcctools.taskvine.compat only works with dask version < 2024.12.0")
-except ImportError:
+except (ImportError, ModuleNotFoundError):
     pass