diff --git a/tests/extract/test_incremental.py b/tests/extract/test_incremental.py index d579d9b9ee..e2681cfbfe 100644 --- a/tests/extract/test_incremental.py +++ b/tests/extract/test_incremental.py @@ -835,14 +835,14 @@ def some_data( assert e.value.json_path == "created_at" # same thing when run in pipeline - with pytest.raises(PipelineStepFailed) as e: + with pytest.raises(PipelineStepFailed) as e: # type: ignore[assignment] p = dlt.pipeline(pipeline_name=uniq_id()) p.extract(some_data()) if item_type == "object": assert isinstance(e.value.__context__, IncrementalCursorPathMissing) else: assert isinstance(e.value.__context__, IncrementalCursorPathHasValueNone) - assert e.value.__context__.json_path == "created_at" + assert e.value.__context__.json_path == "created_at" # type: ignore[attr-defined] @pytest.mark.parametrize("item_type", ["arrow-table", "arrow-batch", "pandas"]) @@ -868,7 +868,7 @@ def some_data( with pytest.raises(PipelineStepFailed) as pip_ex: p = dlt.pipeline(pipeline_name=uniq_id()) p.extract(some_data()) - assert pip_ex.value.__context__.json_path == "created_at" + assert pip_ex.value.__context__.json_path == "created_at" # type: ignore[assignment] assert isinstance(pip_ex.value.__context__, IncrementalCursorPathMissing)