2024-09-04 11:19:30 -07:00
|
|
|
from lib.test_utils import test_load_cmd, test_args
|
|
|
|
|
from .data_types import InputData
|
|
|
|
|
|
|
|
|
|
WORKER_ENDPOINT = "/generate"
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
2024-09-04 11:53:45 -07:00
|
|
|
test_load_cmd(InputData, WORKER_ENDPOINT, arg_parser=test_args)
|
