Commit f0e6627f authored by Siarhei Khaletski (EPAM)'s avatar Siarhei Khaletski (EPAM) 🚩
Browse files

Merge branch 'haaggarw/data-partition' into 'master'

Added support for handling data-partition-id (e.g. CSV parser uses)

See merge request !5
parents 59e58330 9c4bae10
Pipeline #96031 passed with stages
in 2 minutes and 53 seconds
......@@ -109,7 +109,13 @@ class UpdateStatusOperator(BaseOperator):
if "Payload" in execution_context:
payload_context = Context.populate(execution_context)
else:
payload_context = Context(data_partition_id=execution_context["data-partition-id"],
try:
data_partition_id = execution_context["data-partition-id"]
except KeyError:
# to support some DAGs execution_context interface
data_partition_id = execution_context['dataPartitionId']
payload_context = Context(data_partition_id=data_partition_id,
app_key=execution_context.get("AppKey", ""))
workflow_name = conf["workflow_name"]
run_id = conf["run_id"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment