From 312709fdcbb76b4affd17fab6cc836c54b3bda4a Mon Sep 17 00:00:00 2001 From: Gustavo Goretkin Date: Wed, 10 Jan 2024 11:43:20 -0500 Subject: [PATCH] Handle `None` case. --- spot_wrapper/wrapper.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/spot_wrapper/wrapper.py b/spot_wrapper/wrapper.py index 968e88e..406daad 100644 --- a/spot_wrapper/wrapper.py +++ b/spot_wrapper/wrapper.py @@ -851,8 +851,8 @@ def lease(self) -> typing.List[lease_pb2.LeaseResource]: return self._lease_task.proto @property - def lease2(self) -> Lease: - """Return the most recently `take`n or `acquire`d lease""" + def lease2(self) -> typing.Optional[Lease]: + """Return the most recently `take`n or `acquire`d lease, or `None` if it is `release`d.""" return self._lease @property