Update NnforgeWrapper.py