From e080b9b33d6c009d6a141f8dc17e23f9b52091d4 Mon Sep 17 00:00:00 2001 From: Sviatoslav Abakumov Date: Fri, 21 Feb 2025 12:26:42 +0400 Subject: [PATCH] Type hint Lock.extend's additional_time as a Number It must be possible for the function to receive floats, as the docstring and the code suggest. --- redis/lock.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/redis/lock.py b/redis/lock.py index cae7f27ea1..f44ed629da 100644 --- a/redis/lock.py +++ b/redis/lock.py @@ -264,7 +264,7 @@ def do_release(self, expected_token: str) -> None: lock_name=self.name, ) - def extend(self, additional_time: int, replace_ttl: bool = False) -> bool: + def extend(self, additional_time: Number, replace_ttl: bool = False) -> bool: """ Adds more time to an already acquired lock. @@ -281,7 +281,7 @@ def extend(self, additional_time: int, replace_ttl: bool = False) -> bool: raise LockError("Cannot extend a lock with no timeout", lock_name=self.name) return self.do_extend(additional_time, replace_ttl) - def do_extend(self, additional_time: int, replace_ttl: bool) -> bool: + def do_extend(self, additional_time: Number, replace_ttl: bool) -> bool: additional_time = int(additional_time * 1000) if not bool( self.lua_extend(