fix(storage/nvme): use trtype for controller

Signed-off-by: Artsiom Koltun <[email protected]>
opiproject · Nov 29, 2023 · 1a0845d · 1a0845d
1 parent 91579e3
commit 1a0845d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pydpu/storage.py b/pydpu/storage.py
@@ -157,6 +157,7 @@ def create(self, address):
                     nvme_controller_id=str(self.id),
                     nvme_controller=frontend_nvme_pb2.NvmeController(
                         spec=frontend_nvme_pb2.NvmeControllerSpec(
+                            trtype=opicommon_pb2.NVME_TRANSPORT_PCIE,
                             pcie_id=opicommon_pb2.PciEndpoint(
                                 physical_function=wrappers_pb2.Int32Value(
                                     value=self.pf
@@ -288,7 +289,6 @@ def update(self, address):
                     nvme_namespace=frontend_nvme_pb2.NvmeNamespace(
                         name=self.fullname,
                         spec=frontend_nvme_pb2.NvmeNamespaceSpec(
-                            trtype=opicommon_pb2.NVME_TRANSPORT_PCIE,
                             volume_name_ref="Malloc1",
                             uuid=uuid_pb2.Uuid(
                                 value="1b4e28ba-2fa1-11d2-883f-b9a761bde3fb"