From 5a5b9488e8ed7e97e3996931a5a987eafe5d6fd6 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Sat, 25 Sep 2021 23:05:12 +0300
Subject: [PATCH] Add joblib (equivalent to scikit) to Task.init()
 auto_connect_frameworks argument

---
 clearml/task.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/clearml/task.py b/clearml/task.py
index 1f129f75..75f485cb 100644
--- a/clearml/task.py
+++ b/clearml/task.py
@@ -357,7 +357,7 @@ class Task(_Task):
                auto_connect_frameworks={
                    'matplotlib': True, 'tensorflow': True, 'tensorboard': True, 'pytorch': True,
                    'xgboost': True, 'scikit': True, 'fastai': True, 'lightgbm': True,
-                   'hydra': True, 'detect_repository': True, 'tfdefines': True,
+                   'hydra': True, 'detect_repository': True, 'tfdefines': True, 'joblib': True,
                }
 
         :param bool auto_resource_monitoring: Automatically create machine resource monitoring plots
@@ -545,7 +545,9 @@ class Task(_Task):
                 is_auto_connect_frameworks_bool = not isinstance(auto_connect_frameworks, dict)
                 if is_auto_connect_frameworks_bool or auto_connect_frameworks.get('hydra', True):
                     PatchHydra.update_current_task(task)
-                if is_auto_connect_frameworks_bool or auto_connect_frameworks.get('scikit', True):
+                if is_auto_connect_frameworks_bool or (
+                        auto_connect_frameworks.get('scikit', True) and
+                        auto_connect_frameworks.get('joblib', True)):
                     PatchedJoblib.update_current_task(task)
                 if is_auto_connect_frameworks_bool or auto_connect_frameworks.get('matplotlib', True):
                     PatchedMatplotlib.update_current_task(Task.__main_task)