Fix ReifyMetadata (#496)

mlrun · Jan 22, 2024 · d5270f3 · d5270f3
1 parent 28344e0
commit d5270f3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/storey/flow.py b/storey/flow.py
@@ -677,7 +677,7 @@ def __init__(self, mapping: Iterable[str], **kwargs):
     async def _do(self, event):
         if event is not _termination_obj:
             if isinstance(self.mapping, dict):
-                for attribute_name, entry_key in self.mapping:
+                for attribute_name, entry_key in self.mapping.items():
                     event.body[entry_key] = getattr(event, attribute_name)
             else:
                 for attribute_name in self.mapping:

diff --git a/tests/test_flow.py b/tests/test_flow.py
@@ -3860,7 +3860,7 @@ def test_csv_source_event_metadata():
                 timestamp_format="%d/%m/%Y %H:%M:%S",
                 id_field="k",
             ),
-            ReifyMetadata({"key", "id"}),
+            ReifyMetadata({"key": "id"}),
             Reduce([], append_and_return, full_event=False),
         ]
     ).run()
@@ -3872,15 +3872,13 @@ def test_csv_source_event_metadata():
             "b": True,
             "id": "m1",
             "k": "m1",
-            "key": "m1",
             "t": datetime(2020, 2, 15, 2, 0),
             "v": 8,
         },
         {
             "b": False,
             "id": "m2",
             "k": "m2",
-            "key": "m2",
             "t": datetime(2020, 2, 16, 2, 0),
             "v": 14,
         },