ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-03-03 10:42:05 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Replace torch version on pre-installed local file

Browse Source
This commit is contained in:
allegroai 2020-10-04 19:40:39 +03:00
parent 8135a6facf
commit 5640489f57
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -281,7 +281,7 @@ class PytorchRequirement(SimpleSubstitution):
req.name, req.specs[0] if req.specs else 'unspecified', installed_torch[0]['version']))
# package already installed, do nothing
req.specs = [('==', str(installed_torch[0]['version']))]
return str(req), True
return '{} {} {}'.format(req.name, req.specs[0][0], req.specs[0][1]), True
except Exception:
pass
@ -463,6 +463,10 @@ class PytorchRequirement(SimpleSubstitution):
if '@' in line:
# skip if we have nothing to add
if str(req).strip() != str(new_req).strip():
# if this is local file and use the version detection
if req.local_file:
lines[i] = '{}'.format(str(new_req))
else:
lines[i] = '{} # {}'.format(str(req), str(new_req))
else:
lines[i] = '{} # {}'.format(line, str(new_req))