Skip to content

Commit cfb4543

Browse files
fix: strip partitioned_stream_status in model_to_component_factory test assertion
1 parent 98f3710 commit cfb4543

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

unit_tests/sources/declarative/parsers/test_model_to_component_factory.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1327,7 +1327,9 @@ def test_stream_with_incremental_and_async_retriever_with_partition_router(use_l
13271327
assert isinstance(retriever, AsyncRetriever)
13281328
stream_slicer = retriever.stream_slicer.stream_slicer
13291329
assert isinstance(stream_slicer, ConcurrentPerPartitionCursor)
1330-
assert stream_slicer.state == stream_state
1330+
actual_state = stream_slicer.state
1331+
actual_state.pop("partitioned_stream_status", None)
1332+
assert actual_state == stream_state
13311333
import json
13321334

13331335
cursor_perpartition = stream_slicer._cursor_per_partition

0 commit comments

Comments
 (0)