Morgan Funtowicz commited on
Commit
6520a09
·
1 Parent(s): fb05b22

test(hfjobs): various fix

Browse files
Files changed (3) hide show
  1. .hfjobs/Dockerfile +1 -1
  2. Dockerfile +2 -2
  3. tests/test_openai.py +1 -2
.hfjobs/Dockerfile CHANGED
@@ -5,10 +5,10 @@ ARG ENDPOINT_URL="http://host.docker.internal:8000"
5
  ARG ENDPOINT_NUM_SAMPLES=50
6
  ARG ENDPOINT_TEST_SEED=31415926
7
 
8
- ENV ENDPOINT_API_KEY=$ENDPOINT_API_KEY
9
  ENV ENDPOINT_URL=$ENDPOINT_URL
10
  ENV ENDPOINT_NUM_SAMPLES=$ENDPOINT_NUM_SAMPLES
11
  ENV ENDPOINT_TEST_SEED=$ENDPOINT_TEST_SEED
 
12
 
13
  RUN --mount=type=bind,source=.hfjobs/requirements.txt,target=/opt/endpoint/tests/requirements.txt \
14
  python -m pip install -r /opt/endpoint/tests/requirements.txt
 
5
  ARG ENDPOINT_NUM_SAMPLES=50
6
  ARG ENDPOINT_TEST_SEED=31415926
7
 
 
8
  ENV ENDPOINT_URL=$ENDPOINT_URL
9
  ENV ENDPOINT_NUM_SAMPLES=$ENDPOINT_NUM_SAMPLES
10
  ENV ENDPOINT_TEST_SEED=$ENDPOINT_TEST_SEED
11
+ ENV HF_HOME=/tmp/huggingface
12
 
13
  RUN --mount=type=bind,source=.hfjobs/requirements.txt,target=/opt/endpoint/tests/requirements.txt \
14
  python -m pip install -r /opt/endpoint/tests/requirements.txt
Dockerfile CHANGED
@@ -2,10 +2,10 @@ ARG SDK_VERSION=latest
2
  FROM huggingface/hfendpoints-sdk:${SDK_VERSION} AS sdk
3
 
4
  FROM vllm/vllm-openai:v0.8.4
5
- RUN --mount=type=bind,from=sdk,source=/usr/local/hfendpoints/dist,target=/usr/local/hfendpoints/dist \
6
  --mount=type=bind,source=requirements.txt,target=/tmp/requirements.txt \
7
  python3 -m pip install -r /tmp/requirements.txt && \
8
- python3 -m pip install /usr/local/hfendpoints/dist/*.whl
9
 
10
  COPY handler.py /usr/local/endpoint/
11
 
 
2
  FROM huggingface/hfendpoints-sdk:${SDK_VERSION} AS sdk
3
 
4
  FROM vllm/vllm-openai:v0.8.4
5
+ RUN --mount=type=bind,from=sdk,source=/opt/hfendpoints/dist,target=/usr/local/endpoints/dist \
6
  --mount=type=bind,source=requirements.txt,target=/tmp/requirements.txt \
7
  python3 -m pip install -r /tmp/requirements.txt && \
8
+ python3 -m pip install /usr/local/endpoints/dist/*.whl
9
 
10
  COPY handler.py /usr/local/endpoint/
11
 
tests/test_openai.py CHANGED
@@ -7,7 +7,6 @@ import soundfile as sf
7
  import pytest
8
 
9
  # Global variables defining endpoint targets
10
- ENDPOINT_API_KEY = environ["ENDPOINT_API_KEY"]
11
  ENDPOINT_URL = environ["ENDPOINT_URL"]
12
  ENDPOINT_NUM_SAMPLES = int(environ["ENDPOINT_NUM_SAMPLES"])
13
  ENDPOINT_TEST_SEED = int(environ["ENDPOINT_TEST_SEED"])
@@ -20,7 +19,7 @@ from datasets import load_dataset, Dataset
20
  from openai import OpenAI
21
 
22
  # Global client to make requests
23
- client = OpenAI(api_key=ENDPOINT_API_KEY, base_url=ENDPOINT_URL)
24
 
25
 
26
  @pytest.fixture
 
7
  import pytest
8
 
9
  # Global variables defining endpoint targets
 
10
  ENDPOINT_URL = environ["ENDPOINT_URL"]
11
  ENDPOINT_NUM_SAMPLES = int(environ["ENDPOINT_NUM_SAMPLES"])
12
  ENDPOINT_TEST_SEED = int(environ["ENDPOINT_TEST_SEED"])
 
19
  from openai import OpenAI
20
 
21
  # Global client to make requests
22
+ client = OpenAI(base_url=ENDPOINT_URL)
23
 
24
 
25
  @pytest.fixture