From 3649eb8d978fdade1cfe0dd1d1ba413e8d58578f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Galip=20=C3=9Cmit=20Yolcu?=
 <galip.uemit.yolcu@hhi.fraunhofer.de>
Date: Thu, 13 Jun 2024 19:18:45 +0200
Subject: [PATCH] return ranking instead of self-influence values

---
 src/explainers/aggregators/self_influence.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/explainers/aggregators/self_influence.py b/src/explainers/aggregators/self_influence.py
index b0f83a14..e7be7367 100644
--- a/src/explainers/aggregators/self_influence.py
+++ b/src/explainers/aggregators/self_influence.py
@@ -29,3 +29,4 @@ def get_self_influence_ranking(
             train_ids=[i],
             **explain_fn_kwargs,
         )
+    return self_inf.argsort()