Add additional warning instructing on how to install in case we faile…

…d detecting jupyter notebook with an import error
allegroai · Apr 19, 2024 · 3d920a4 · 3d920a4
1 parent 2717c7e
commit 3d920a4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/clearml/backend_interface/task/repo/scriptinfo.py b/clearml/backend_interface/task/repo/scriptinfo.py
@@ -348,6 +348,12 @@ def _daemon(cls, jupyter_notebook_filename, notebook_name=None):
 
         except Exception as ex:
             cls._get_logger().warning('Could not read Jupyter Notebook: {}'.format(ex))
+            if isinstance(ex, ImportError):
+                module_name = getattr(ex, "name", None)
+                if module_name:
+                    cls._get_logger().warning(
+                        'Please install {name} using "pip install {name}"'.format(name=module_name)
+                    )
             _script_exporter = None
 
         # load pigar