Skip to content

Commit 6e13ad2

Browse files
committed
fix: suppress mypy errors in new GCS file_data tests
1 parent fb49ae6 commit 6e13ad2

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

tests/unittests/artifacts/test_artifact_service.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -870,10 +870,10 @@ def test_converts_text_dict(self):
870870
# ---------------------------------------------------------------------------
871871

872872

873-
@pytest.mark.asyncio
874-
async def test_gcs_save_artifact_with_external_gcs_uri():
873+
@pytest.mark.asyncio # type: ignore[untyped-decorator]
874+
async def test_gcs_save_artifact_with_external_gcs_uri() -> None:
875875
"""GcsArtifactService saves and loads a gs:// file_data URI reference."""
876-
service = mock_gcs_artifact_service()
876+
service = mock_gcs_artifact_service() # type: ignore[no-untyped-call]
877877
artifact = types.Part(
878878
file_data=types.FileData(
879879
file_uri="gs://my-bucket/report.pdf",
@@ -902,10 +902,10 @@ async def test_gcs_save_artifact_with_external_gcs_uri():
902902
assert loaded.file_data.mime_type == "application/pdf"
903903

904904

905-
@pytest.mark.asyncio
906-
async def test_gcs_save_artifact_with_artifact_ref_uri():
905+
@pytest.mark.asyncio # type: ignore[untyped-decorator]
906+
async def test_gcs_save_artifact_with_artifact_ref_uri() -> None:
907907
"""GcsArtifactService saves and loads an internal artifact:// URI reference."""
908-
service = mock_gcs_artifact_service()
908+
service = mock_gcs_artifact_service() # type: ignore[no-untyped-call]
909909
artifact_ref_uri = "artifact://apps/app/users/user1/sessions/sess1/artifacts/source.txt/versions/0"
910910
artifact = types.Part(
911911
file_data=types.FileData(
@@ -934,10 +934,10 @@ async def test_gcs_save_artifact_with_artifact_ref_uri():
934934
assert loaded.file_data.file_uri == artifact_ref_uri
935935

936936

937-
@pytest.mark.asyncio
938-
async def test_gcs_save_artifact_file_data_without_mime_type():
937+
@pytest.mark.asyncio # type: ignore[untyped-decorator]
938+
async def test_gcs_save_artifact_file_data_without_mime_type() -> None:
939939
"""GcsArtifactService handles file_data with no mime_type."""
940-
service = mock_gcs_artifact_service()
940+
service = mock_gcs_artifact_service() # type: ignore[no-untyped-call]
941941
artifact = types.Part(
942942
file_data=types.FileData(file_uri="gs://my-bucket/data.bin")
943943
)
@@ -962,10 +962,10 @@ async def test_gcs_save_artifact_file_data_without_mime_type():
962962
assert loaded.file_data.file_uri == "gs://my-bucket/data.bin"
963963

964964

965-
@pytest.mark.asyncio
966-
async def test_gcs_save_artifact_file_data_missing_uri_raises():
965+
@pytest.mark.asyncio # type: ignore[untyped-decorator]
966+
async def test_gcs_save_artifact_file_data_missing_uri_raises() -> None:
967967
"""GcsArtifactService raises InputValidationError when file_uri is empty."""
968-
service = mock_gcs_artifact_service()
968+
service = mock_gcs_artifact_service() # type: ignore[no-untyped-call]
969969
artifact = types.Part(file_data=types.FileData(file_uri=""))
970970

971971
with pytest.raises(InputValidationError):

0 commit comments

Comments
 (0)