fix format

vllm-project · Jan 6, 2025 · 0a3d315 · 0a3d315
1 parent 28869b0
commit 0a3d315
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git 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
@@ -1,5 +1,3 @@
-import os
-
 from vllm import LLM, SamplingParams
 
 # Sample prompts.