diff --git a/dockerfile/ubuntu-20.04-amd64.Dockerfile b/dockerfile/ubuntu-20.04-amd64.Dockerfile index 25d4e710094..c934a6bc7c9 100644 --- a/dockerfile/ubuntu-20.04-amd64.Dockerfile +++ b/dockerfile/ubuntu-20.04-amd64.Dockerfile @@ -34,6 +34,8 @@ ENV PYTHON_ENV_DIR=${TT_METAL_INFRA_DIR}/tt-metal/python_env # Copy requirements from tt-metal folders with requirements.txt docs COPY /docs/requirements-docs.txt ${TT_METAL_INFRA_DIR}/tt-metal/docs/. +# Copy requirements from tt-metal folders for sweeps (requirements-sweeps.txt) +COPY /tests/sweep_framework/requirements-sweeps.txt ${TT_METAL_INFRA_DIR}/tt-metal/tests/sweep_framework/. COPY /tt_metal/python_env/* ${TT_METAL_INFRA_DIR}/tt-metal/tt_metal/python_env/. RUN python3 -m pip config set global.extra-index-url https://download.pytorch.org/whl/cpu \ && python3 -m pip install setuptools wheel