def register(cli):
    version_info = device_info.get_sonic_version_info()
    if (version_info and (version_info.get('asic_type') == 'mellanox')):
        cli.commands['platform'].add_command(mlnx)
def register(cli):
    version_info = device_info.get_sonic_version_info()
    if (version_info and (version_info.get('asic_type') == 'mellanox')):
        cli.commands['platform'].add_command(mlnx)