From a605ec8e04636b6decb9be047879570bfd388658 Mon Sep 17 00:00:00 2001 From: Diogo Sousa Date: Mon, 3 Jul 2023 13:44:13 +0100 Subject: [PATCH] add todo --- .../openbb_sdk_core/openbb_sdk_core/app/utils.py | 16 +--------------- 1 file changed, 1 insertion(+), 15 deletions(-) diff --git a/openbb_sdk/openbb_sdk_core/openbb_sdk_core/app/utils.py b/openbb_sdk/openbb_sdk_core/openbb_sdk_core/app/utils.py index 5dee360e3ea..1f02aee8628 100644 --- a/openbb_sdk/openbb_sdk_core/openbb_sdk_core/app/utils.py +++ b/openbb_sdk/openbb_sdk_core/openbb_sdk_core/app/utils.py @@ -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