import os import sys from nni.retiarii.integration import RetiariiAdvisor