Fix 1793 (#1794)

Co-authored-by: Daniel Girtler <girtler.daniel@gmail.com>
archlinux · May 7, 2023 · 525c94b · 525c94b
1 parent eb23b2d
commit 525c94b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/archinstall/lib/disk/device_handler.py b/archinstall/lib/disk/device_handler.py
@@ -10,7 +10,7 @@
 from parted import (  # type: ignore
 	Disk, Geometry, FileSystem,
 	PartitionException, DiskLabelException,
-	getAllDevices, freshDisk, Partition,
+	getAllDevices, freshDisk, Partition, Device
 )
 
 from .device_model import (
@@ -103,7 +103,8 @@ def get_device(self, path: Path) -> Optional[BDevice]:
 	def get_device_by_partition_path(self, partition_path: Path) -> Optional[BDevice]:
 		partition = self.find_partition(partition_path)
 		if partition:
-			return partition.disk.device
+			device: Device = partition.disk.device
+			return self.get_device(Path(device.path))
 		return None
 
 	def find_partition(self, path: Path) -> Optional[_PartitionInfo]: