chore: add mypy strict type hints for partition module

swysocki · Jul 31, 2022 · 90d17f9 · 90d17f9
1 parent 93ddf50
commit 90d17f9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gpt_image/partition.py b/gpt_image/partition.py
@@ -98,7 +98,7 @@ def attribute_flags(self) -> List[int]:
         return flags
 
     @attribute_flags.setter
-    def attribute_flags(self, flag: PartitionAttribute):
+    def attribute_flags(self, flag: PartitionAttribute) -> None:
         """Set the partition attribute flag
 
         Sets the bit corresponding to the PartitionAttribute Class. If the
@@ -207,7 +207,7 @@ def _get_first_lba(self, partition: Partition) -> int:
             integer of the first LBA
         """
 
-        def next_lba(end_lba: int, alignment: int):
+        def next_lba(end_lba: int, alignment: int) -> int:
             m = int(end_lba / alignment)
             return (m + 1) * alignment