ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 00:56:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Auto detect tensorflow-intel as tensorflow (there is no tensorflow-intel package)

Browse Source
This commit is contained in:
allegroai 2024-01-06 12:32:57 +02:00
parent 4bd6b20a62
commit 0ed5ff4123
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
clearml/backend_interface/task/repo/scriptinfo.py
View File

@ -97,6 +97,11 @@ class ScriptRequirements(object):
for fname, lines in skimage.items():
modules.add('scikit_image', fname, lines)
if 'tensorflow-intel' in modules:
tfmodule = modules.pop('tensorflow-intel', {})
for fname, lines in tfmodule.items():
modules.add('tensorflow', fname, lines)
# if we have torch and it supports tensorboard, we should add that as well
# (because it will not be detected automatically)
if 'torch' in modules and 'tensorboard' not in modules and 'tensorboardX' not in modules: