Skip to content

Commit b8f1ebb

Browse files
committed
Fixed mypy errors
1 parent b79289f commit b8f1ebb

File tree

3 files changed

+6
-4
lines changed

3 files changed

+6
-4
lines changed

src/redis_anyio/_connection.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,9 @@ async def aclose(self, *, force: bool = True) -> None:
106106
else:
107107
await self._send_stream.aclose()
108108

109-
async def run(self, stream: AnyByteStream, *, task_status: TaskStatus) -> None:
109+
async def run(
110+
self, stream: AnyByteStream, *, task_status: TaskStatus[None]
111+
) -> None:
110112
send, self._response_stream = create_memory_object_stream(100)
111113
async with stream, send:
112114
self._send_stream = stream

src/redis_anyio/_lock.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ async def _run_script(
8282
await self.client.script_load(source)
8383
return await self.client.evalsha(script_sha1, [self.name], list(args))
8484

85-
async def _run(self, *, task_status: TaskStatus) -> None:
85+
async def _run(self, *, task_status: TaskStatus[None]) -> None:
8686
while not await self._run_script(
8787
ACQUIRE_SCRIPT, ACQUIRE_SCRIPT_SHA, self._token, self.lifetime
8888
):

tests/test_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -587,7 +587,7 @@ class TestLock:
587587
async def test_locking(self, redis_port: int, separate_locks: bool) -> None:
588588
events: list[str] = []
589589

590-
async def acquire_lock(*, task_status: TaskStatus) -> None:
590+
async def acquire_lock(*, task_status: TaskStatus[None]) -> None:
591591
async with lock1:
592592
task_status.started()
593593
events.append("subtask acquired the lock")
@@ -633,7 +633,7 @@ async def test_task_holding_lock_cancelled(
633633
one from getting the lock.
634634
"""
635635

636-
async def acquire_lock(*, task_status: TaskStatus) -> None:
636+
async def acquire_lock(*, task_status: TaskStatus[None]) -> None:
637637
async with lock:
638638
task_status.started()
639639
await sleep(5)

0 commit comments

Comments
 (0)