Dynamically add disks to vm based on input parameters as number of disks and size of disks

[yobshans]

Hello,
I found example https://github.com/nutanix/terraform-provider-nutanix/tree/master/examples/virtual_machine_with_dynamic_disk_list which is pretty good.
But it based on parameter type = list(string) default = [1024,2048].
I have a little bit different configuration and would like to pass 2 parameters which are:
master_extra_disk_count and master_extra_disk_size (numbers).
Trying to use count meta-argument such as
disk_list {
count = var.master_extra_disk_count
disk_size_bytes = master_extra_disk_size
device_properties {
device_type = "DISK"
disk_address {
device_index = count.index
adapter_type = "SATA"
}
}
}
But got error
│ Error: Unsupported argument
│
│ on virtual_machines.tf line 25, in resource "nutanix_virtual_machine" "master":
│ 25: count = var.master_extra_disk_count
│
│ An argument named "count" is not expected here.
Could you recommend any workaround ?

Thank you