fixes

pympipool/interactive/dependencies.py

@@ -3,15 +3,7 @@
 from pympipool.interactive import create_executor
 from pympipool.shared.executor import ExecutorSteps, execute_tasks_with_dependencies
 from pympipool.shared.thread import RaisingThread
-
-
-try:
-    from pympipool.shared.plot import generate_task_hash, generate_nodes_and_edges, draw
-
-    graph_extension_available = True
-    import_error = None
-except ImportError:
-    graph_extension_available = False
+from pympipool.shared.plot import generate_task_hash, generate_nodes_and_edges, draw
 
 
 class ExecutorWithDependencies(ExecutorSteps):
@@ -38,8 +30,6 @@ def __init__(
         )
         self._future_hash_dict = {}
         self._task_hash_dict = {}
-        if plot_dependency_graph and not graph_extension_available:
-            raise ImportError(import_error)
         self._generate_dependency_graph = plot_dependency_graph
 
     def submit(self, fn: callable, *args, resource_dict: dict = {}, **kwargs):

pympipool/shared/plot.py

@@ -1,9 +1,6 @@
 from concurrent.futures import Future
 
 import cloudpickle
-from IPython.display import SVG, display
-import matplotlib.pyplot as plt
-import networkx as nx
 
 
 def generate_nodes_and_edges(task_hash_dict, future_hash_inverse_dict):
@@ -59,6 +56,9 @@ def generate_task_hash(task_dict, future_hash_inverse_dict):
 
 
 def draw(node_lst, edge_lst):
+    from IPython.display import SVG, display  # noqa
+    import matplotlib.pyplot as plt  # noqa
+    import networkx as nx  # noqa
     graph = nx.DiGraph()
     for node in node_lst:
         graph.add_node(node["id"], label=node["name"])