added node index to current_context

Signed-off-by: Atharva <[email protected]>

Author: Atharva1723

flytekit/bin/entrypoint.py

@@ -470,7 +470,9 @@ def setup_execution(
     if checkpoint_path is not None:
         checkpointer = SyncCheckpoint(checkpoint_dest=checkpoint_path, checkpoint_src=prev_checkpoint)
         logger.debug(f"Checkpointer created with source {prev_checkpoint} and dest {checkpoint_path}")
-
+    
+    node_index = _compute_array_job_index()
+    
     execution_parameters = ExecutionParameters(
         execution_id=_identifier.WorkflowExecutionIdentifier(
             project=exe_project,
@@ -498,6 +500,7 @@ def setup_execution(
         output_metadata_prefix=output_metadata_prefix,
         checkpoint=checkpointer,
         task_id=_identifier.Identifier(_identifier.ResourceType.TASK, tk_project, tk_domain, tk_name, tk_version),
+        node_index=node_index,
     )
 
     metadata = {