diff --git a/common/sai_npu.py b/common/sai_npu.py index 6f51bf16..cf364876 100644 --- a/common/sai_npu.py +++ b/common/sai_npu.py @@ -212,9 +212,8 @@ def set_sku_mode(self, sku): self.remove(oid) self.remove(self.dot1q_bp_oids[idx]) status, data = self.get(self.port_oids[idx], ["SAI_PORT_ATTR_PORT_SERDES_ID"], do_assert=False) - serdes_oid = data.oid() - if status == "SAI_STATUS_SUCCESS" and serdes_oid != "oid:0x0": - self.remove(serdes_oid) + if status == "SAI_STATUS_SUCCESS" and data.oid() != "oid:0x0": + self.remove(data.oid()) self.remove(self.port_oids[idx]) self.port_oids.clear() self.dot1q_bp_oids.clear()