Files
pyworker/workers/tgi/test_load.py
T
Nader Arbabian 589216d15f initial commit
2024-09-04 11:19:30 -07:00

8 lines
224 B
Python

from lib.test_utils import test_load_cmd, test_args
from .data_types import InputData
WORKER_ENDPOINT = "/generate"
if __name__ == "__main__":
test_load_cmd(InputData.for_test(), WORKER_ENDPOINT, arg_parser=test_args)