try this using ansible-pull

roles/base/tasks/archinstall/initialize_root_zvol.yaml

@@ -18,6 +18,9 @@
         }}
     with_items: "{{ base_partitions_by_id | dict2items }}"
 
+  - debug:
+      var: base_root_partitions | map('extract', partition_device_to_ids) | map('first')
+
   - name: Create zroot volume
     become: true
     command: >-
@@ -32,7 +35,7 @@
         -O canmount=off
         -O mountpoint=none
         -R /mnt
-        zroot {{ mirror }} {{ base_root_partitions | map('extract', partition_device_to_ids) | flatten | list | join(' ') }}
+        zroot {{ mirror }} {{ base_root_partitions | map('extract', partition_device_to_ids) | map('first') | flatten | list | join(' ') }}
     vars:
       mirror: "{{ 'mirror' if base_root_partitions | length > 1 else '' }}"
   when: zroot_check.rc != 0

roles/base/tasks/archinstall/partition.yaml

@@ -69,9 +69,24 @@
   loop: "{{ base_root_disks_info.results }}"
   register: parted_create
 
-- name: Store EFI partition details
+- name: Collect EFI partition devices
+  command: "lsblk -r --noheadings -o PATH {{ item }}"
+  register: lsblk
+  loop: "{{ parted_create.results | map(attribute='disk.dev') }}"
+
+- debug:
+    var: lsblk
+
+- name: Store EFI partition devices
   set_fact:
-    base_efi_partitions: "{{ parted_create.results | map(attribute='disk.dev') | product([1]) | map('join') | list }}"
+    base_efi_partitions: >-
+      {{ base_efi_partitions|d([]) + [
+          item[part_index+1]
+        ]
+      }}
+  loop: "{{ lsblk.results | map(attribute='stdout_lines') }}"
+  vars:
+    part_index: 0
 
 - name: Create root zvol partition
   become: true
@@ -89,9 +104,21 @@
   loop: "{{ parted_create.results }}"
   register: parted_create
 
-- name: Store root zvol partition details
+- name: Collect root zvol partition devices
+  command: "lsblk -r --noheadings -o PATH {{ item }}"
+  register: lsblk
+  loop: "{{ parted_create.results | map(attribute='disk.dev') }}"
+
+- name: Store EFI partition devices
   set_fact:
-    base_root_partitions: "{{ parted_create.results | map(attribute='disk.dev') | product([2]) | map('join') | list }}"
+    base_root_partitions: >-
+      {{ base_root_partitions|d([]) + [
+          item[part_index+1]
+        ]
+      }}
+  loop: "{{ lsblk.results | map(attribute='stdout_lines') }}"
+  vars:
+    part_index: 1
 
 - name: Create swap partition
   become: true
@@ -108,12 +135,23 @@
     fs_type: linux-swap
   loop: "{{ parted_create.results }}"
   register: parted_create
-  when: base_root_swap_mib is defined and base_root_swap_mib > 0
+  when: base_root_swap_mib is defined and base_root_swap_mib|int > 0
 
-- name: Store swap partition details
+- name: Collect swap partition devices
+  command: "lsblk -r --noheadings -o PATH {{ item }}"
+  register: lsblk
+  loop: "{{ parted_create.results | map(attribute='disk.dev') }}"
+
+- name: Store EFI partition devices
   set_fact:
-    base_swap_partitions: "{{ parted_create.results | map(attribute='disk.dev') | product([3]) | map('join') | list }}"
-  when: base_root_swap_mib is defined and base_root_swap_mib > 0
+    base_swap_partitions: >-
+      {{ base_swap_partitions|d([]) + [
+          item[part_index+1]
+        ]
+      }}
+  loop: "{{ lsblk.results | map(attribute='stdout_lines') }}"
+  vars:
+    part_index: 2
 
 - name: Analyze resulting partition layouts
   parted:
@@ -136,3 +174,7 @@
           | map('list')
         )
       }}
+
+- debug: var=base_efi_partitions
+- debug: var=base_root_partitions
+- debug: var=base_swap_partitions

roles/base/templates/arch_chroot_install.sh

@@ -56,8 +56,7 @@ function configure_zfs_initrd {
 	# Add 'zfs' to MODULES=()
 	sed -i '/^MODULES=/!b;/zfs/!s/)/zfs)/' /etc/mkinitcpio.conf
 	initrd=/boot/initramfs-linux-lts.img
-	[[ ! -e $initrd || $initrd -ot /etc/mkinitcpio.conf ]] || mkinitcpio -P
-	# ^ only rebuild if initrd image is older than mkinitpio.conf
+	[[ ! -e "$initrd" || "$initrd" -ot /etc/mkinitcpio.conf ]] && mkinitcpio -P
 }
 
 function configure_dhcp_ethernet {