diff --git a/.buildkite/run-neuron-test.sh b/.buildkite/run-neuron-test.sh index 9b978365ce793..aa29c434e7cfb 100644 --- a/.buildkite/run-neuron-test.sh +++ b/.buildkite/run-neuron-test.sh @@ -27,8 +27,8 @@ if [ -f /tmp/neuron-docker-build-timestamp ]; then if [ $((current_time - last_build)) -gt 86400 ]; then docker image prune -f docker system prune -f - rm -rf "${HF_MOUNT}/*" - rm -rf "${NEURON_COMPILE_CACHE_MOUNT}/*" + rm -rf "${HF_MOUNT:?}/*" + rm -rf "${NEURON_COMPILE_CACHE_MOUNT:?}/*" echo "$current_time" > /tmp/neuron-docker-build-timestamp fi else diff --git a/examples/offline_inference_neuron.py b/examples/offline_inference_neuron.py index 9079c44a68242..f098c8e5fed1e 100644 --- a/examples/offline_inference_neuron.py +++ b/examples/offline_inference_neuron.py @@ -1,5 +1,3 @@ -import os - from vllm import LLM, SamplingParams # Sample prompts.