ML-4879: Avoid keeping reference to events in AggregateByKey when c…

…ache is disabled (#463) Fixes [ML-4879](https://jira.iguazeng.com/browse/ML-4879). Co-authored-by: Gal Topper <[email protected]>
mlrun · Nov 2, 2023 · 1ede8e3 · 1ede8e3
1 parent ca426ae
commit 1ede8e3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/storey/table.py b/storey/table.py
@@ -190,7 +190,8 @@ async def _aggregate(self, key, event, data, timestamp):
             cache_item = self._get_aggregations_attrs(key)
             await cache_item.aggregate(data, timestamp)
             self._changed_keys.add(key)
-        self._pending_events.append(event)
+        if self._flush_task:
+            self._pending_events.append(event)
 
     async def _get_features(self, key, timestamp):
         if self._flush_exception is not None: