4545
4646
4747def stage_files_for_operator (
48- operator , request_id = None , pairing = None , job_group_id = None , job_group_notifier_id = None , parent = None , notify = False
48+ operator ,
49+ request_id = None ,
50+ pairing = None ,
51+ file_group_id = None ,
52+ job_group_id = None ,
53+ job_group_notifier_id = None ,
54+ parent = None ,
55+ notify = False ,
4956):
5057 staging_tasks = []
5158 try :
@@ -62,6 +69,7 @@ def stage_files_for_operator(
6269 operator_id = operator ,
6370 request_id = request_id ,
6471 pairing = pairing ,
72+ file_group_id = file_group_id ,
6573 job_group_id = job_group_id ,
6674 job_group_notifier_id = job_group_notifier_id ,
6775 parent = parent ,
@@ -75,6 +83,7 @@ def stage_files_for_operator(
7583 operator_id = operator ,
7684 request_id = request_id ,
7785 pairing = pairing ,
86+ file_group_id = file_group_id ,
7887 job_group_id = job_group_id ,
7988 job_group_notifier_id = job_group_notifier_id ,
8089 parent = parent ,
@@ -88,6 +97,7 @@ def create_operator_run_from_jobs(
8897 request_id = None ,
8998 pairing = None ,
9099 run_ids = None ,
100+ file_group_id = None ,
91101 job_group_id = None ,
92102 job_group_notifier_id = None ,
93103 parent = None ,
@@ -97,6 +107,7 @@ def create_operator_run_from_jobs(
97107 operator_model = Operator .objects .get (id = operator_id )
98108 operator = OperatorFactory .get_by_model (
99109 operator_model ,
110+ file_group = file_group_id ,
100111 job_group_id = job_group_id ,
101112 job_group_notifier_id = job_group_notifier_id ,
102113 request_id = request_id ,
@@ -284,7 +295,7 @@ def stage_files(request_id=None, pairing=None, job_group_id=None):
284295@shared_task
285296@tracer .wrap (service = "beagle" )
286297def create_jobs_from_request (
287- request_id , operator_id , job_group_id , job_group_notifier_id = None , pipeline = None , file_group = None , notify = False
298+ request_id , operator_id , job_group_id , job_group_notifier_id = None , file_group = None , pipeline = None , notify = False
288299):
289300 current_span = tracer .current_span ()
290301 current_span .set_tag ("request.id" , request_id )
@@ -337,6 +348,7 @@ def create_jobs_from_request(
337348 job_group_id = job_group_id ,
338349 job_group_notifier_id = job_group_notifier_id ,
339350 notify = notify ,
351+ file_group_id = file_group ,
340352 )
341353
342354
@@ -369,10 +381,10 @@ def create_jobs_from_pairs(
369381 operator = OperatorFactory .get_by_model (
370382 operator_model ,
371383 pairing = {"pairs" : pairs },
384+ file_group = file_group_id ,
372385 job_group_id = job_group_id ,
373386 job_group_notifier_id = job_group_notifier_id ,
374387 request_id = request_id ,
375- file_group = file_group_id ,
376388 output_directory_prefix = output_directory_prefix ,
377389 )
378390
@@ -383,6 +395,7 @@ def create_jobs_from_pairs(
383395 pairing = {"pairs" : pairs },
384396 job_group_id = job_group_id ,
385397 job_group_notifier_id = job_group_notifier_id ,
398+ file_group_id = file_group_id ,
386399 )
387400
388401
0 commit comments