fix ut issue

Signed-off-by: Dong, Bo1 <[email protected]>
opea-project · Sep 3, 2024 · eb3ef45 · eb3ef45
1 parent 747827e
commit eb3ef45
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/comps/reranks/neural-speed/neuralspeed-docker/Dockerfile b/comps/reranks/neural-speed/neuralspeed-docker/Dockerfile
@@ -7,9 +7,9 @@ ARG DEBIAN_FRONTEND=noninteractive
 ENV GLIBC_TUNABLES glibc.cpu.x86_shstk=permissive
 
 COPY comps /root/comps
-COPY /data2/nswhl/neural_speed-0.1.dev45+g41ea0aa-cp310-cp310-linux_x86_64.whl /root/
-COPY /data2/nswhl/bge-large-r-q8.bin /root/
-COPY /data2/nswhl/libstdc++.so.6 /root/
+COPY neural_speed-0.1.dev45+g41ea0aa-cp310-cp310-linux_x86_64.whl /root/
+COPY bge-large-r-q8.bin /root/
+COPY libstdc++.so.6 /root/
 
 RUN apt update && apt install -y python3 python3-pip 
 RUN pip3 install -r /root/comps/reranks/neural-speed/neuralspeed-docker/requirements.txt

diff --git a/tests/test_reranks_mosec-neuralspeed.sh b/tests/test_reranks_mosec-neuralspeed.sh
@@ -10,6 +10,7 @@ ip_address=$(hostname -I | awk '{print $1}')
 function build_mosec_docker_images() {
     cd $WORKPATH
     echo $(pwd)
+    cp /data2/nswhl/* ./
     docker build --build-arg http_proxy=$http_proxy --build-arg https_proxy=$https_proxy -t langchain-mosec:neuralspeed-reranks -f comps/reranks/neural-speed/neuralspeed-docker/Dockerfile .
     if [ $? -ne 0 ]; then
         echo "opea/reranking-langchain-mosec-endpoint built fail"