Replace foreach stack with foreach indices

Netflix · Jan 11, 2025 · 27386bb · 27386bb
1 parent c9ffc19
commit 27386bb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/metaflow/client/core.py b/metaflow/client/core.py
@@ -1180,7 +1180,7 @@ def _get_metadata_query_vals(
             print(f"Query task is {query_task}")
             print(f"Query task metadata is {query_task.metadata_dict}")
             query_foreach_stack_len = len(
-                query_task.metadata_dict.get("foreach-stack", [])
+                query_task.metadata_dict.get("foreach-indices", [])
             )
 
         if query_foreach_stack_len == cur_foreach_stack_len:
@@ -1241,7 +1241,8 @@ def _get_related_tasks(self, relation_type: str) -> Dict[str, List[str]]:
         field_name, field_value = self._get_metadata_query_vals(
             flow_id,
             run_id,
-            len(self.metadata_dict.get("foreach-stack", [])),
+            len(self.metadata_dict.get("foreach-indices", [])),
+            # len(self.metadata_dict.get("foreach-stack", [])),
             steps,
             relation_type,
         )