gen_partition.py: add NVMe as a valid disk type

NVMe flashing follows the same process as eMMC. Add `nvme` as a valid
disk type to explicitly call out the supported storage options.

Signed-off-by: Viswanath Kraleti <[email protected]>

Author: vkraleti

gen_partition.py

@@ -211,7 +211,7 @@ def generate_nand_xml (disk_params, partition_entries_dict, output_xml):
 def generate_partition_xml (disk_entry, partition_entries_dict, output_xml):
    parse_disk_entry(disk_entry)
    print("Generating %s XML %s" %(disk_params["type"].upper(), output_xml))
-   if disk_params["type"] == "emmc":
+   if disk_params["type"] in ("emmc", "nvme"):
       generate_emmc_xml(disk_params, partition_entries_dict, output_xml)
    elif disk_params["type"] == "nand":
       generate_nand_xml(disk_params, partition_entries_dict, output_xml)