From 0f0f5db3eb27e57f74c7028a45832d1c796d41c7 Mon Sep 17 00:00:00 2001 From: Younes Strittmatter Date: Sat, 10 Aug 2024 14:18:41 -0400 Subject: [PATCH] feat: add alias reference_conditions for X and reference_observations for y --- src/autora/state.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/src/autora/state.py b/src/autora/state.py index e6bf16f3..c4bbcb1d 100644 --- a/src/autora/state.py +++ b/src/autora/state.py @@ -1506,6 +1506,13 @@ def X(self) -> pd.DataFrame: return pd.DataFrame() return self.experiment_data[self.iv_names] + @property + def reference_conditions(self) -> pd.DataFrame: + """ + Returns the already observed conditions as a pd.DataFrame. + """ + return self.X + @property def y(self) -> pd.DataFrame: """ @@ -1533,6 +1540,13 @@ def y(self) -> pd.DataFrame: return pd.DataFrame() return self.experiment_data[self.dv_names] + @property + def reference_observations(self) -> pd.DataFrame: + """ + Returns the observations as a pd.DataFrame + """ + return self.y + @property def model(self): if len(self.models) == 0: