We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 98f3710 commit cfb4543Copy full SHA for cfb4543
1 file changed
unit_tests/sources/declarative/parsers/test_model_to_component_factory.py
@@ -1327,7 +1327,9 @@ def test_stream_with_incremental_and_async_retriever_with_partition_router(use_l
1327
assert isinstance(retriever, AsyncRetriever)
1328
stream_slicer = retriever.stream_slicer.stream_slicer
1329
assert isinstance(stream_slicer, ConcurrentPerPartitionCursor)
1330
- assert stream_slicer.state == stream_state
+ actual_state = stream_slicer.state
1331
+ actual_state.pop("partitioned_stream_status", None)
1332
+ assert actual_state == stream_state
1333
import json
1334
1335
cursor_perpartition = stream_slicer._cursor_per_partition
0 commit comments