Update metadata API params

Netflix · Nov 1, 2024 · ec43f14 · ec43f14
1 parent 0ef1087
commit ec43f14
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/metaflow/metadata/metadata.py b/metaflow/metadata/metadata.py
@@ -674,16 +674,16 @@ def _register_system_metadata(self, run_id, step_name, task_id, attempt):
 
     @classmethod
     def _filter_tasks_by_metadata(
-        cls, flow_id, run_id, step_name, prev_step, field_name, field_value
+        cls, flow_id, run_id, step_name, query_step, field_name, field_value
     ):
         raise NotImplementedError()
 
     @classmethod
     def filter_tasks_by_metadata(
-        cls, flow_id, run_id, step_name, prev_step, field_name, field_value
+        cls, flow_id, run_id, step_name, query_step, field_name, field_value
     ):
         task_ids = cls._filter_tasks_by_metadata(
-            flow_id, run_id, step_name, prev_step, field_name, field_value
+            flow_id, run_id, step_name, query_step, field_name, field_value
         )
         return task_ids