diff --git a/sonic_platform_base/module_base.py b/sonic_platform_base/module_base.py index 25136ada9..95af3c606 100644 --- a/sonic_platform_base/module_base.py +++ b/sonic_platform_base/module_base.py @@ -262,26 +262,26 @@ def get_state_info(self): """ raise NotImplementedError - def get_dpu_bus_info(self, dpu_name): + def get_bus_info(self, module_name): """ - Retrieves the DPU bus information for the specified by "dpu_name" on a SmartSwitch. + Retrieves the bus information for the specified by "module_name" on a SmartSwitch. Returns: Returns the PCI bus information in BDF format like "[DDDD:]BB:SS:F" """ raise NotImplementedError - def pci_detach_dpu(self, dpu_name): + def pci_detach_dpu(self, module_name): """ - Detaches the DPU PCI device specified by "dpu_name" on a SmartSwitch. + Detaches the DPU PCI device specified by "module_name" on a SmartSwitch. Returns: True once the PCI is successfully detached. """ raise NotImplementedError - def pci_reattach_dpu(self, dpu_name): + def pci_reattach_dpu(self, module_name): """ - Rescans and reconnects the DPU PCI device specified by "dpu_name" on a SmartSwitch. + Rescans and reconnects the DPU PCI device specified by "module_name" on a SmartSwitch. Returns: True once the PCI is successfully reconnected. """ diff --git a/tests/module_base_test.py b/tests/module_base_test.py index 16b53eff1..5d39f4997 100644 --- a/tests/module_base_test.py +++ b/tests/module_base_test.py @@ -8,9 +8,9 @@ def test_module_base(self): [module.get_dpu_id], [module.get_reboot_cause], [module.get_state_info], - [module.get_dpu_bus_info, "dpu_name"], - [module.pci_detach_dpu, "dpu_name"], - [module.pci_reattach_dpu, "dpu_name"], + [module.get_bus_info, "module_name"], + [module.pci_detach_dpu, "module_name"], + [module.pci_reattach_dpu, "module_name"], ] for method in not_implemented_methods: