Files
pyworker/workers/tgi/test_load.py
Nader Arbabian ed7a952bf4 Merge pull request #1 from Nader-gator/main
add pyworker v2
2024-09-12 11:27:48 -07:00

8 lines
213 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, WORKER_ENDPOINT, arg_parser=test_args)