From 0ed5ff41238e384ad7b75457a20e1d5cea1bc71b Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Sat, 6 Jan 2024 12:32:57 +0200 Subject: [PATCH] Auto detect tensorflow-intel as tensorflow (there is no tensorflow-intel package) --- clearml/backend_interface/task/repo/scriptinfo.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/clearml/backend_interface/task/repo/scriptinfo.py b/clearml/backend_interface/task/repo/scriptinfo.py index f9c08a82..c8f21f77 100644 --- a/clearml/backend_interface/task/repo/scriptinfo.py +++ b/clearml/backend_interface/task/repo/scriptinfo.py @@ -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: