Pin setuptools to fix pytorch example failures (#5255)

* Pin setuptools to fix pytorch examples failures Signed-off-by: harupy <17039389+harupy@users.noreply.github.com> * todo Signed-off-by: harupy <17039389+harupy@users.noreply.github.com>
mlflow · Jan 12, 2022 · c2ae0bd · c2ae0bd
1 parent 9362410
commit c2ae0bd
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 0 deletions.
diff --git a/examples/pytorch/BertNewsClassification/conda.yaml b/examples/pytorch/BertNewsClassification/conda.yaml
@@ -14,3 +14,10 @@ dependencies:
   - torch>=1.9.0
   - torchtext==0.10.0
   - pytorch-lightning==1.5.1
+  # torch <= 1.10.1 doesn't work with setuptools >= 59.6.0.
+  # See the following PRs for details:
+  # https://github.com/pytorch/pytorch/pull/69904
+  # https://github.com/pytorch/pytorch/pull/69947
+  #
+  # TODO: Remove this requirement once torch > 1.10.1 is released.
+  - setuptools<59.6.0
diff --git a/examples/pytorch/IterativePruning/conda.yaml b/examples/pytorch/IterativePruning/conda.yaml
@@ -11,3 +11,10 @@ dependencies:
   - ax-platform
   - prettytable
   - torch>=1.9.0
+  # torch <= 1.10.1 doesn't work with setuptools >= 59.6.0.
+  # See the following PRs for details:
+  # https://github.com/pytorch/pytorch/pull/69904
+  # https://github.com/pytorch/pytorch/pull/69947
+  #
+  # TODO: Remove this requirement once torch > 1.10.1 is released.
+  - setuptools<59.6.0
diff --git a/examples/pytorch/MNIST/conda.yaml b/examples/pytorch/MNIST/conda.yaml
@@ -8,3 +8,10 @@ dependencies:
   - torchvision>=0.9.1
   - torch>=1.9.0
   - pytorch-lightning==1.5.1
+  # torch <= 1.10.1 doesn't work with setuptools >= 59.6.0.
+  # See the following PRs for details:
+  # https://github.com/pytorch/pytorch/pull/69904
+  # https://github.com/pytorch/pytorch/pull/69947
+  #
+  # TODO: Remove this requirement once torch > 1.10.1 is released.
+  - setuptools<59.6.0