diff --git a/.buildkite/test-pipeline.yaml b/.buildkite/test-pipeline.yaml index d7d2b930e6e7a..b788002c1ee9b 100644 --- a/.buildkite/test-pipeline.yaml +++ b/.buildkite/test-pipeline.yaml @@ -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 diff --git a/examples/llava_example.py b/examples/llava_example.py index a455e98585983..3d22b492654bf 100644 --- a/examples/llava_example.py +++ b/examples/llava_example.py @@ -78,7 +78,13 @@ def main(args): # 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)