@@ -152,7 +152,7 @@ def pytask_collect_task(session, path, name, obj):
152152
153153
154154@hookspec
155- def pytask_collect_task_teardown (task ):
155+ def pytask_collect_task_teardown (session , task ):
156156 """Perform tear-down operations when a task was collected.
157157
158158 Use this hook specification to, for example, perform checks on the collected task.
@@ -161,7 +161,7 @@ def pytask_collect_task_teardown(task):
161161
162162
163163@hookspec (firstresult = True )
164- def pytask_collect_node (path , node ):
164+ def pytask_collect_node (session , path , node ):
165165 """Collect a node which is a dependency or a product of a task."""
166166
167167
@@ -207,7 +207,7 @@ def pytask_resolve_dependencies(session):
207207
208208
209209@hookspec (firstresult = True )
210- def pytask_resolve_dependencies_create_dag (tasks ):
210+ def pytask_resolve_dependencies_create_dag (session , tasks ):
211211 """Create the DAG.
212212
213213 This hook creates the DAG from tasks, dependencies and products. The DAG can be used
@@ -217,7 +217,7 @@ def pytask_resolve_dependencies_create_dag(tasks):
217217
218218
219219@hookspec (firstresult = True )
220- def pytask_resolve_dependencies_select_execution_dag (dag ):
220+ def pytask_resolve_dependencies_select_execution_dag (session , dag ):
221221 """Select the subgraph which needs to be executed.
222222
223223 This hook determines which of the tasks have to be re-run because something has
@@ -227,7 +227,7 @@ def pytask_resolve_dependencies_select_execution_dag(dag):
227227
228228
229229@hookspec (firstresult = True )
230- def pytask_resolve_dependencies_validate_dag (dag ):
230+ def pytask_resolve_dependencies_validate_dag (session , dag ):
231231 """Validate the DAG.
232232
233233 This hook validates the DAG. For example, there can be cycles in the DAG if tasks,
0 commit comments