base_name check

Signed-off-by: Shmulik Froimovich <[email protected]>
IBM · Jul 4, 2023 · 7dcfc67 · 7dcfc67
1 parent cae6397
commit 7dcfc67
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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}')