text-generation-webui/modules/relative_imports.py
2023-06-19 12:31:24 -03:00

14 lines
308 B
Python

import sys
from pathlib import Path
class RelativeImport:
def __init__(self, path):
self.import_path = Path(path)
def __enter__(self):
sys.path.insert(0, str(self.import_path))
def __exit__(self, exc_type, exc_value, traceback):
sys.path.remove(str(self.import_path))