Morgan Funtowicz commited on
Commit
12f816d
·
1 Parent(s): fe7a3eb

test(hfjobs): numba cache folder

Browse files
Files changed (1) hide show
  1. .hfjobs/Dockerfile +3 -1
.hfjobs/Dockerfile CHANGED
@@ -5,10 +5,12 @@ ARG ENDPOINT_URL="http://host.docker.internal:8000"
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
 
5
  ARG ENDPOINT_NUM_SAMPLES=50
6
  ARG ENDPOINT_TEST_SEED=31415926
7
 
8
+ ENV NUMBA_CACHE_DIR=/tmp/numba
9
+ ENV HF_HOME=/tmp/huggingface
10
+
11
  ENV ENDPOINT_URL=$ENDPOINT_URL
12
  ENV ENDPOINT_NUM_SAMPLES=$ENDPOINT_NUM_SAMPLES
13
  ENV ENDPOINT_TEST_SEED=$ENDPOINT_TEST_SEED
 
14
 
15
  RUN --mount=type=bind,source=.hfjobs/requirements.txt,target=/opt/endpoint/tests/requirements.txt \
16
  python -m pip install -r /opt/endpoint/tests/requirements.txt