do not overwrite index (#79)

logix-project · Jan 8, 2024 · 2f373cf · 2f373cf
1 parent 5c123bf
commit 2f373cf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/analog/analysis/influence_function.py b/analog/analysis/influence_function.py
@@ -79,7 +79,8 @@ def compute_influence(
             assert total_influence.shape[1] == len(tgt_ids)
             # Ensure src_ids and tgt_ids are in the DataFrame's index and columns, respectively
             self.influence_scores = self.influence_scores.reindex(
-                index=src_ids, columns=tgt_ids
+                index=self.influence_scores.index.union(src_ids),
+                columns=self.influence_scores.columns.union(tgt_ids),
             )
 
             # Assign total_influence values to the corresponding locations in influence_scores