Skip to content

Commit

Permalink
Update
Browse files Browse the repository at this point in the history
  • Loading branch information
xyang16 committed Jul 24, 2024
1 parent 268c44c commit de54e6b
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion tests/integration/llm/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -907,14 +907,18 @@ def awscurl_run(data,
if tokenizer:
command = f"TOKENIZER={tokenizer} {command}"
if output:
output_path = os.path.join(os.path.curdir, "outputs", "output")
output_dir = os.path.join(os.path.curdir, "outputs")
os.mkdir(output_dir)
output_path = os.path.join(output_dir, "output")
command = f"{command} -o {output_path}"
LOGGER.info(f"Running command {command}")
res = sp.run(command, shell=True, stdout=sp.PIPE, stderr=sp.PIPE)
if "error rate: 100" in res.stdout.decode("utf-8"):
raise ValueError("Found error result in awscurl_run")
if dataset:
shutil.rmtree(dataset_dir)
if output:
shutil.rmtree(output_dir)


def send_image_json(img_url, data):
Expand Down

0 comments on commit de54e6b

Please sign in to comment.