diff --git a/packages/py/aitbc-sdk/tests/test_receipts.py b/packages/py/aitbc-sdk/tests/test_receipts.py index 7b892ebb..281a9e66 100755 --- a/packages/py/aitbc-sdk/tests/test_receipts.py +++ b/packages/py/aitbc-sdk/tests/test_receipts.py @@ -194,8 +194,9 @@ def test_iter_receipts_handles_pagination(monkeypatch, sample_payload: Dict[str, def test_request_retries_on_transient(monkeypatch, sample_payload: Dict[str, object]) -> None: + from aitbc import NetworkError responses: List[object] = [ - httpx.ReadTimeout("timeout"), + NetworkError("timeout"), _DummyResponse(429, {}), _DummyResponse(200, {}), ]