do not import DaskVineDag if dask not available

cooperative-computing-lab · Jan 8, 2025 · 5ed5dcb · 5ed5dcb
1 parent 283b036
commit 5ed5dcb
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/taskvine/src/bindings/python3/ndcctools/taskvine/__init__.py b/taskvine/src/bindings/python3/ndcctools/taskvine/__init__.py
@@ -57,12 +57,11 @@
     LibraryTask,
     FunctionCall,
 )
-from .dask_dag import DaskVineDag
-
 from . import cvine
 
 try:
     from .dask_executor import DaskVine
+    from .dask_dag import DaskVineDag
 except ImportError as e:
     print(f"DaskVine not available. Couldn't find module: {e.name}")