diff --git a/.github/workflows/test_inf2.yml b/.github/workflows/test_inf2.yml index c68e30a59..567aa8980 100644 --- a/.github/workflows/test_inf2.yml +++ b/.github/workflows/test_inf2.yml @@ -41,6 +41,7 @@ jobs: python3 -m venv aws_neuron_venv_pytorch source aws_neuron_venv_pytorch/bin/activate python -m pip install -U pip + pip install --upgrade setuptools==69.5.1 python -m pip config set global.extra-index-url https://pip.repos.neuron.amazonaws.com python -m pip install .[neuronx,tests] - name: Run cache tests diff --git a/.github/workflows/test_inf2_export.yml b/.github/workflows/test_inf2_export.yml index 91af90317..d0bebb778 100644 --- a/.github/workflows/test_inf2_export.yml +++ b/.github/workflows/test_inf2_export.yml @@ -41,6 +41,7 @@ jobs: python3 -m venv aws_neuron_venv_pytorch source aws_neuron_venv_pytorch/bin/activate python -m pip install -U pip + pip install --upgrade setuptools==69.5.1 python -m pip config set global.extra-index-url https://pip.repos.neuron.amazonaws.com python -m pip install .[neuronx,tests] - name: Run exporters tests diff --git a/.github/workflows/test_inf2_full_export.yml b/.github/workflows/test_inf2_full_export.yml index 9025ee3ef..b16274920 100644 --- a/.github/workflows/test_inf2_full_export.yml +++ b/.github/workflows/test_inf2_full_export.yml @@ -39,6 +39,7 @@ jobs: python3 -m venv aws_neuron_venv_pytorch source aws_neuron_venv_pytorch/bin/activate python -m pip install -U pip + pip install --upgrade setuptools==69.5.1 python -m pip config set global.extra-index-url https://pip.repos.neuron.amazonaws.com python -m pip install .[neuronx,tests] - name: Run exporters tests diff --git a/.github/workflows/test_inf2_inference.yml b/.github/workflows/test_inf2_inference.yml index 62bed4a32..006a73110 100644 --- a/.github/workflows/test_inf2_inference.yml +++ b/.github/workflows/test_inf2_inference.yml @@ -41,6 +41,7 @@ jobs: python3 -m venv aws_neuron_venv_pytorch source aws_neuron_venv_pytorch/bin/activate python -m pip install -U pip + pip install --upgrade setuptools==69.5.1 python -m pip config set global.extra-index-url https://pip.repos.neuron.amazonaws.com python -m pip install .[neuronx,tests] - name: Run inference tests diff --git a/pyproject.toml b/pyproject.toml index f337ddb32..70fd7db8d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. [build-system] -requires = ["setuptools>=69.5.1"] +requires = ["setuptools==69.5.1"] build-backend = "setuptools.build_meta" [tool.black]