ARG tag_prefix=latest
FROM gcr.io/tensorflow-sigs/build-arm64:tf-${tag_prefix}-multi-python

ARG py_major_minor_version='3.10'

ENV PYTHON_BIN_PATH=/usr/bin/python${py_major_minor_version}

RUN ln -s ${PYTHON_BIN_PATH} /usr/local/bin/python && \
    ln -s ${PYTHON_BIN_PATH} /usr/local/bin/python3

COPY upper-constraints.txt /
RUN ${PYTHON_BIN_PATH} -m pip install -c /upper-constraints.txt -r /upper-constraints.txt

RUN mkdir -p /tf/xla
RUN chown buildslave:buildslave /tf/xla