more syntactical sugar

OSInside · Jan 18, 2024 · 9ad8a51 · 9ad8a51
1 parent b519ff1
commit 9ad8a51
Showing 1 changed file with 19 additions and 13 deletions.
diff --git a/kiwi/builder/disk.py b/kiwi/builder/disk.py
@@ -74,22 +74,28 @@
     KiwiVolumeManagerSetupError
 )
 
-StorageMap = TypedDict(
-    'StorageMap', {
-        'system': Optional[Union[FileSystemBase, VolumeManagerBase]],
-        'system_boot': Optional[FileSystemBase],
-        'system_efi': Optional[FileSystemBase],
-        'system_spare': Optional[FileSystemBase],
-        'system_custom_parts': Dict[str, FileSystemBase],
-        'luks_root': Optional[LuksDevice],
-        'raid_root': Optional[RaidDevice],
-        'integrity_root': Optional[IntegrityDevice]
-    }
-)
-
 log = logging.getLogger('kiwi')
 
 
+class StorageMap(TypedDict):
+    system: \
+        Optional[Union[FileSystemBase, VolumeManagerBase]]
+    system_boot: \
+        Optional[FileSystemBase]
+    system_efi: \
+        Optional[FileSystemBase]
+    system_spare: \
+        Optional[FileSystemBase]
+    system_custom_parts: \
+        Dict[str, FileSystemBase]
+    luks_root: \
+        Optional[LuksDevice]
+    raid_root: \
+        Optional[RaidDevice]
+    integrity_root: \
+        Optional[IntegrityDevice]
+
+
 class DiskBuilder:
     """
     **Disk image builder**