From 666c81262a0324c2031a5a574bd6c6f6ad390f99 Mon Sep 17 00:00:00 2001
From: Amir MOHAMMADI <amir.mohammadi@idiap.ch>
Date: Wed, 18 Nov 2020 14:07:30 +0100
Subject: [PATCH] [repodata-patches] Add dataclasses and future to Pytorch Also
remove the cpuonly patch from Pytorch packages
---
deps/repodata-patches/gen_patch_json.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/deps/repodata-patches/gen_patch_json.py b/deps/repodata-patches/gen_patch_json.py
index 65dee25c..90d6ae3a 100644
--- a/deps/repodata-patches/gen_patch_json.py
+++ b/deps/repodata-patches/gen_patch_json.py
@@ -118,10 +118,11 @@ def _gen_new_index(repodata, packages_key):
if parse_version(record["version"]) <= parse_version("4.1.0"):
record["depends"].append("numpy <1.18")
- # somehow conda cannot resolve pytorch cpu without the cpuonly package
- # we only ship cpu-only pytorch packages
+ # pytorch 1.7.0 0 packages need this extra dependencies to run
if record_name == "pytorch":
- record["depends"].append("cpuonly")
+ if record["version"] == "1.7.0" and record["build_number"] == 0:
+ record["depends"].append("dataclasses >=0.8")
+ record["depends"].append("future >=0.18.2")
return index
--
GitLab