polish code

flagscale/runner/runner_serve.py

@@ -81,7 +81,7 @@ def _generate_run_script_serve(
     root_dir = os.path.dirname(
         os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
     )
-    vllm_dir = os.path.join(root_dir, "vllm")
+    # vllm_dir = os.path.join(root_dir, "vllm")
     cmds_config = config.experiment.get("cmds", None)
     if cmds_config:
         before_start = cmds_config.get("before_start", "")
@@ -97,7 +97,7 @@ def _generate_run_script_serve(
         f.write(f"\n")
         f.write(f"cd {root_dir}\n")
         f.write(f"\n")
-        f.write(f"export PYTHONPATH={vllm_dir}:{root_dir}\n")
+        f.write(f"export PYTHONPATH={root_dir}\n")
         f.write(f"\n")
         f.write(f'cmd="{cmd}"\n')
         f.write(f"\n")

tests/scripts/functional_tests/test_model.sh

@@ -82,21 +82,6 @@ test_model() {
         if [ $? -ne 0 ]; then
           echo "Test failed on attempt $attempt_i for case $_case."
 
-          FILE="./outputs/multiple_model/serve_logs/host_0_localhost.output"
-
-          if [ -e "$FILE" ]; then
-              echo "log exists: $FILE"
-              echo "-----------------print log -----------------------"
-              cat "$FILE"
-              echo "------------------curl call----------------------"
-              ifconfig
-              env
-              # curl -v http://127.0.0.1:6701/generate -H "Content-Type: application/json" -d '{"prompt": "introduce Bruce Lee2"}'
-              echo "------------------curl result----------------------"
-          else
-              echo "log not exists: $FILE"
-          fi
-
           clear_serve
           exit 1
         fi