We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 74a046c commit a064295Copy full SHA for a064295
1 file changed
src/decoupler/bm/_run.py
@@ -50,7 +50,9 @@ def _tensor_scores(
50
51
def _tensor_truth(obs: pd.DataFrame, srcs: np.ndarray) -> pd.DataFrame:
52
# Explode nested perturbs and pivot into mat
53
- grts = obs.explode("source").pivot(columns="source", values="type_p").notna().astype(float).fillna(0.0).loc[obs.index]
+ grts = (
54
+ obs.explode("source").pivot(columns="source", values="type_p").notna().astype(float).fillna(0.0).loc[obs.index]
55
+ )
56
miss_srcs = srcs[~np.isin(srcs, grts.columns)]
57
miss_srcs = pd.DataFrame(0, index=grts.index, columns=miss_srcs)
58
grts = pd.concat([grts, miss_srcs], axis=1)
0 commit comments