Handle None case.

bdaiinstitute · Jan 10, 2024 · 312709f · 312709f
1 parent 5a593cb
commit 312709f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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