@@ -152,7 +152,7 @@ def pytask_collect_task(session, path, name, obj):
152
152
153
153
154
154
@hookspec
155
- def pytask_collect_task_teardown (task ):
155
+ def pytask_collect_task_teardown (session , task ):
156
156
"""Perform tear-down operations when a task was collected.
157
157
158
158
Use this hook specification to, for example, perform checks on the collected task.
@@ -161,7 +161,7 @@ def pytask_collect_task_teardown(task):
161
161
162
162
163
163
@hookspec (firstresult = True )
164
- def pytask_collect_node (path , node ):
164
+ def pytask_collect_node (session , path , node ):
165
165
"""Collect a node which is a dependency or a product of a task."""
166
166
167
167
@@ -207,7 +207,7 @@ def pytask_resolve_dependencies(session):
207
207
208
208
209
209
@hookspec (firstresult = True )
210
- def pytask_resolve_dependencies_create_dag (tasks ):
210
+ def pytask_resolve_dependencies_create_dag (session , tasks ):
211
211
"""Create the DAG.
212
212
213
213
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):
217
217
218
218
219
219
@hookspec (firstresult = True )
220
- def pytask_resolve_dependencies_select_execution_dag (dag ):
220
+ def pytask_resolve_dependencies_select_execution_dag (session , dag ):
221
221
"""Select the subgraph which needs to be executed.
222
222
223
223
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):
227
227
228
228
229
229
@hookspec (firstresult = True )
230
- def pytask_resolve_dependencies_validate_dag (dag ):
230
+ def pytask_resolve_dependencies_validate_dag (session , dag ):
231
231
"""Validate the DAG.
232
232
233
233
This hook validates the DAG. For example, there can be cycles in the DAG if tasks,
0 commit comments