import os import sys # add current folder to the python path sys.path.append(os.path.dirname(__file__)) # this is required to dynamically load the PIE models from pie_modules.models import * # noqa: F403 from pie_modules.taskmodules import * # noqa: F403 from pytorch_ie.models import * # noqa: F403 from pytorch_ie.taskmodules import * # noqa: F403