Skip to content

Commit 74a046c

Browse files
Fix index retention in _tensor_truth function
Ensure the grts retains original index after pivoting. Currently shuffles grout truth tensor compared to activities inferred by decouple
1 parent 6e18630 commit 74a046c

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/decoupler/bm/_run.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def _tensor_scores(
5050

5151
def _tensor_truth(obs: pd.DataFrame, srcs: np.ndarray) -> pd.DataFrame:
5252
# Explode nested perturbs and pivot into mat
53-
grts = obs.explode("source").pivot(columns="source", values="type_p").notna().astype(float).fillna(0.0)
53+
grts = obs.explode("source").pivot(columns="source", values="type_p").notna().astype(float).fillna(0.0).loc[obs.index]
5454
miss_srcs = srcs[~np.isin(srcs, grts.columns)]
5555
miss_srcs = pd.DataFrame(0, index=grts.index, columns=miss_srcs)
5656
grts = pd.concat([grts, miss_srcs], axis=1)

0 commit comments

Comments
 (0)