From 7dcfc671c976c684f0128d9b057b4ca0faf9d6e7 Mon Sep 17 00:00:00 2001 From: Shmulik Froimovich Date: Tue, 4 Jul 2023 10:10:38 +0300 Subject: [PATCH] base_name check Signed-off-by: Shmulik Froimovich --- nca/Utils/ExplTracker.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/nca/Utils/ExplTracker.py b/nca/Utils/ExplTracker.py index 34a92d0c..e03646bd 100644 --- a/nca/Utils/ExplTracker.py +++ b/nca/Utils/ExplTracker.py @@ -336,9 +336,9 @@ def prepare_node_str(self, node_name, results, direction=None): if not self.ExplDescriptorContainer.get(name): out.append(f'{ep_name} - explainability entry not found') continue - base_name = self.ExplDescriptorContainer.get(name).get("base_name") - if base_name: - ep_name = base_name + # base_name = self.ExplDescriptorContainer.get(name).get("base_name") + # if base_name: + # ep_name = base_name path = self.ExplDescriptorContainer.get(name).get("path") if path == '': # special element (like Default Policy) out.append(f'{ep_name}')