test / tests /conftest.py
iblfe's picture
Upload folder using huggingface_hub
b585c7f verified
import os
import sys
import importlib.util
def pytest_itemcollected(item):
item._nodeid = item.nodeid + os.getenv("PYTEST_TEST_NAME", "")
def pytest_sessionstart(session):
if not os.getenv("USE_WHEEL", None):
return
try:
for location in importlib.util.find_spec("h2ogpt").submodule_search_locations:
sys.path.append(location)
except AttributeError:
pass