Fix test (#500)

[ML-5665](https://jira.iguazeng.com/browse/ML-5665)
mlrun · Feb 5, 2024 · c38a6bb · c38a6bb
1 parent b369026
commit c38a6bb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_flow.py b/tests/test_flow.py
@@ -231,11 +231,11 @@ def test_offset_commit_before_termination():
             event.offset = offset
             controller.emit(event)
 
+    del event
+
     time.sleep(max_wait_before_commit + 1)
 
     expected_offsets = {("/", i): num_records_per_shard for i in range(num_shards)}
-    # TODO: Remove when commit of last record is fixed
-    expected_offsets[("/", 9)] = 9
 
     try:
         offsets = copy.copy(platform.offsets)