Skip to content

Commit 2638d30

Browse files
committed
Fix checkstyle
1 parent 96b29a0 commit 2638d30

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

graphdatascience/tests/unit/arrow_client/V2/test_write_back_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ def mock_arrow_client(mocker: MockerFixture) -> AuthenticatedArrowClient:
1414
client = mocker.Mock(spec=AuthenticatedArrowClient)
1515
client.connection_info.return_value = mocker.Mock(host="localhost", port=8080, encrypted=False)
1616
client.request_token.return_value = "test_token"
17-
return client
17+
return client # type: ignore
1818

1919

2020
@pytest.fixture
@@ -53,5 +53,5 @@ def test_write_calls_run_write_back(write_back_client: WriteBackClient, mocker:
5353

5454
duration = write_back_client.write(graph_name, job_id, concurrency)
5555

56-
write_back_client._write_protocol.run_write_back.assert_called_once()
56+
write_back_client._write_protocol.run_write_back.assert_called_once() # type: ignore
5757
assert duration >= 0

0 commit comments

Comments
 (0)