ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-06-26 18:16:15 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix torch support to not change back the same link

Browse Source
This commit is contained in:
allegroai 2020-10-04 19:37:12 +03:00
parent 075736de20
commit a56f032ec4
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
trains_agent/helper/package/pytorch.py
View File

@ -461,6 +461,8 @@ class PytorchRequirement(SimpleSubstitution):
if req.req.name == parts[0]: if req.req.name == parts[0]:
# support for pip >= 20.1 # support for pip >= 20.1
if '@' in line: if '@' in line:
# skip if we have nothing to add
if str(req).strip() != str(new_req).strip():
lines[i] = '{} # {}'.format(str(req), str(new_req)) lines[i] = '{} # {}'.format(str(req), str(new_req))
else: else:
lines[i] = '{} # {}'.format(line, str(new_req)) lines[i] = '{} # {}'.format(line, str(new_req))