Update schema.py

microsoft · Nov 23, 2024 · 96a0f9d · 96a0f9d
1 parent c172b96
commit 96a0f9d
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/lisa/schema.py b/lisa/schema.py
@@ -432,18 +432,18 @@ class DiskType(str, Enum):
 
 
 class StorageInterfaceType(str, Enum):
-    NVME = "NVMe"
-    SCSI = "SCSI"
+    SCSI = constants.STORAGE_INTERFACE_TYPE_SCSI
+    NVME = constants.STORAGE_INTERFACE_TYPE_NVME
 
 
 class DiskControllerType(str, Enum):
-    SCSI = StorageInterfaceType.SCSI
-    NVME = StorageInterfaceType.NVME
+    SCSI = constants.STORAGE_INTERFACE_TYPE_SCSI
+    NVME = constants.STORAGE_INTERFACE_TYPE_NVME
 
 
 class ResourceDiskType(str, Enum):
-    SCSI = StorageInterfaceType.SCSI
-    NVME = StorageInterfaceType.NVME
+    SCSI = constants.STORAGE_INTERFACE_TYPE_SCSI
+    NVME = constants.STORAGE_INTERFACE_TYPE_NVME
 
 
 disk_controller_type_priority: List[DiskControllerType] = [