| 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 | |