fix: Fix vllm finder module path (#4691)

Signed-off-by: tzulingk@nvidia.com <tzulingk@nvidia.com>

tests/fault_tolerance/deploy/container/Dockerfile.local_vllm

@@ -105,12 +105,12 @@ COPY --from=vllm_source /usr/local/lib/python3.12/dist-packages ${VIRTUAL_ENV}/l
 RUN if [ -f ${VIRTUAL_ENV}/lib/python${PYTHON_VERSION}/site-packages/__editable__.pplx_kernels-0.0.1.pth ]; then \
         echo "/opt/vllm/ep_kernels_workspace/pplx-kernels/src" > ${VIRTUAL_ENV}/lib/python${PYTHON_VERSION}/site-packages/__editable__.pplx_kernels-0.0.1.pth; \
     fi && \
-    # Fix ALL DeepEP .pth files (version-agnostic using find since version includes git hash)
-    find ${VIRTUAL_ENV}/lib/python${PYTHON_VERSION}/site-packages -name "__editable__.deep_ep*.pth" -exec sed -i 's|/vllm-workspace|/opt/vllm|g' {} \; && \
-    # Fix DeepEP finder modules (these contain the actual hardcoded MAPPING paths)
-    find ${VIRTUAL_ENV}/lib/python${PYTHON_VERSION}/site-packages -name "__editable___deep_ep*_finder.py" -exec sed -i "s|'/vllm-workspace|'/opt/vllm|g" {} \; && \
-    # Also check for any other .pth files that might reference /vllm-workspace
-    find ${VIRTUAL_ENV}/lib/python${PYTHON_VERSION}/site-packages -name "*.pth" -exec sed -i 's|/vllm-workspace|/opt/vllm|g' {} \;
+    # Fix ALL editable install paths (DeepEP .pth files and finder modules for both DeepEP and vLLM)
+    find ${VIRTUAL_ENV}/lib/python${PYTHON_VERSION}/site-packages \
+        \( -name "__editable__.deep_ep*.pth" -o \
+           -name "__editable___deep_ep*_finder.py" -o \
+           -name "__editable___vllm*_finder.py" \) \
+        -exec sed -i "s|/vllm-workspace|/opt/vllm|g" {} \; && \
 
 # Set LD_LIBRARY_PATH for all components
 ENV LD_LIBRARY_PATH=\