try this using ansible-pull

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