jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-29 13:32:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

add todo

Browse Source
This commit is contained in:
Diogo Sousa
2023-07-03 13:44:13 +01:00
parent 66c2f0d3e5
commit a605ec8e04
1 changed files with 1 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
openbb_sdk/openbb_sdk_core/openbb_sdk_core/app/utils.py
View File

@@ -23,18 +23,4 @@ def df_to_basemodel(
return [BaseModel(**d) for d in df.to_dict(orient="records")]
def get_target_column(df: pd.DataFrame, target: str) -> pd.Series:
"""Get target column from time series data."""
if target not in df.columns:
choices = ", ".join(df.columns)
raise ValueError(
f"Target column '{target}' not found in data. Choose from {choices}"
)
return df[target]
def get_target_columns(df: pd.DataFrame, target_columns: List[str]) -> pd.DataFrame:
df_result = pd.DataFrame()
for target in target_columns:
df_result[target] = get_target_column(df, target).to_frame()
return df_result
# TODO: Move utils in common used in ta and qa to here