mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-09-18 23:35:45 +00:00
14 lines
308 B
Python
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))
|