Unverified Commit a4075cba authored by Simon Mo's avatar Simon Mo Committed by GitHub
Browse files

[CI] Add test case to run examples scripts (#3638)

parent 96aa014d
......@@ -36,6 +36,16 @@ steps:
- label: Entrypoints Test
command: pytest -v -s entrypoints
- label: Examples Test
working_dir: "/vllm-workspace/examples"
commands:
# install aws cli for llava_example.py
- pip install awscli
- python3 offline_inference.py
- python3 offline_inference_with_prefix.py
- python3 llm_engine_example.py
- python3 llava_example.py
- label: Kernels Test %N
command: pytest -v -s kernels --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
parallelism: 4
......
......@@ -78,7 +78,13 @@ if __name__ == "__main__":
# Make sure the local directory exists or create it
os.makedirs(local_directory, exist_ok=True)
# Use AWS CLI to sync the directory
subprocess.check_call(
["aws", "s3", "sync", s3_bucket_path, local_directory])
# Use AWS CLI to sync the directory, assume anonymous access
subprocess.check_call([
"aws",
"s3",
"sync",
s3_bucket_path,
local_directory,
"--no-sign-request",
])
main(args)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment