From 7f0f7bb139dd426fb5aba4f096395817c70090b7 Mon Sep 17 00:00:00 2001
From: c-bata <contact@c-bata.link>
Date: Fri, 6 Jan 2023 10:14:52 +0900
Subject: [PATCH] Fix target arg for importance

---
 optuna_dashboard/_importance.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/optuna_dashboard/_importance.py b/optuna_dashboard/_importance.py
index 98f22aea..efc62b41 100644
--- a/optuna_dashboard/_importance.py
+++ b/optuna_dashboard/_importance.py
@@ -96,7 +96,9 @@ def get_param_importance_from_trials_cache(
             return cache_importance
 
         study = StudyWrapper(storage, study_id, trials)
-        importance = _get_param_importances(study, completed_trials)
+        importance = _get_param_importances(
+            study, completed_trials, target=lambda t: t.values[objective_id]
+        )
         converted = convert_to_importance_type(importance, trials)
         param_importance_cache[cache_key] = (n_completed_trials, converted)
     return converted