Skip to content

Commit c60d911

Browse files
committed
enh - transform yhat and logp to pandas dataframe
1 parent a859cb5 commit c60d911

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

pcntoolkit/dataio/norm_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1105,7 +1105,7 @@ def to_dataframe(self, dim_order: Sequence[Hashable] | None = None) -> pd.DataFr
11051105
"""
11061106
acc = []
11071107
x_columns = [col for col in ["X"] if hasattr(self, col)]
1108-
y_columns = [col for col in ["Y", "Y_harmonized", "Z"] if hasattr(self, col)]
1108+
y_columns = [col for col in ["Y", "Y_harmonized", "Z", "logp", "Yhat"] if hasattr(self, col)]
11091109
acc.append(
11101110
xr.Dataset.to_dataframe(self[x_columns], dim_order)
11111111
.reset_index(drop=False)

0 commit comments

Comments
 (0)