modify mellanox.py