Skip to content

Commit

Permalink
fix: pin setuptools version
Browse files Browse the repository at this point in the history
  • Loading branch information
dacorvo committed May 23, 2024
1 parent 0e04e53 commit d4cdf77
Show file tree
Hide file tree
Showing 8 changed files with 11 additions and 4 deletions.
1 change: 1 addition & 0 deletions .github/workflows/test_inf2.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 1 addition & 0 deletions .github/workflows/test_inf2_export.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 1 addition & 0 deletions .github/workflows/test_inf2_full_export.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 1 addition & 0 deletions .github/workflows/test_inf2_inference.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 1 addition & 0 deletions .github/workflows/test_trainium_common.yml
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,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: Collect tests on Neuron Cores
Expand Down
1 change: 1 addition & 0 deletions .github/workflows/test_trainium_distributed.yml
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,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: Collect tests on Neuron Cores
Expand Down
7 changes: 4 additions & 3 deletions .github/workflows/test_trainium_examples.yml
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@ name: Optimum Neuron - Test Example Scripts

on:
workflow_dispatch:
inputs:
coverage:
inputs:
coverage:
description: Coverage
type: choice
options:
Expand All @@ -12,7 +12,7 @@ on:
- middle
- low
required: true
model_size:
model_size:
description: Size of models
type: choice
options:
Expand Down Expand Up @@ -53,6 +53,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: Collect example tests on Neuron Cores
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand Down

0 comments on commit d4cdf77

Please sign in to comment.