Bug fixing

automl · Aug 29, 2024 · e14d90c · e14d90c
1 parent 38588b1
commit e14d90c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepcave/evaluators/mo_fanova.py b/deepcave/evaluators/mo_fanova.py
@@ -153,6 +153,6 @@ def get_importances_(
             raise RuntimeError("Importance scores must be calculated first.")
 
         if hp_names:
-            return self.importances_.loc[self.importances_["hp_name"].isin(hp_names)]
+            return self.importances_.loc[self.importances_["hp_name"].isin(hp_names)].to_json()
         else:
             return self.importances_.to_json()