Fix logic error in _fetch_lsblk_info() loop

archlinux · Oct 29, 2023 · 2077ff5 · 2077ff5
1 parent 2831c97
commit 2077ff5
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/archinstall/lib/disk/device_model.py b/archinstall/lib/disk/device_model.py
@@ -1106,10 +1106,7 @@ def _fetch_lsblk_info(dev_path: Optional[Union[Path, str]] = None, retry: int =
 	if not dev_path:
 		dev_path = ''
 
-	if retry == 0:
-		retry = 1
-
-	for retry_attempt in range(retry):
+	for attempt in range(retry + 1):
 		try:
 			result = SysCommand(f'lsblk --json -b -o+{lsblk_fields} {dev_path}').decode()
 			break
@@ -1121,7 +1118,7 @@ def _fetch_lsblk_info(dev_path: Optional[Union[Path, str]] = None, retry: int =
 			else:
 				raise err
 
-			if retry_attempt == retry - 1:
+			if attempt == retry:
 				raise err
 
 			time.sleep(1)