From d79da5fae16f0c91e8a34d508ce79f7e3eca5b8b Mon Sep 17 00:00:00 2001 From: Diogo Sousa Date: Wed, 5 Jul 2023 18:21:34 +0100 Subject: [PATCH] add to_dict method --- .../openbb_sdk_core/app/model/command_output.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/openbb_sdk/openbb_sdk_core/openbb_sdk_core/app/model/command_output.py b/openbb_sdk/openbb_sdk_core/openbb_sdk_core/app/model/command_output.py index 089beb6028f..64e9ec27628 100644 --- a/openbb_sdk/openbb_sdk_core/openbb_sdk_core/app/model/command_output.py +++ b/openbb_sdk/openbb_sdk_core/openbb_sdk_core/app/model/command_output.py @@ -52,7 +52,7 @@ class CommandOutput(GenericModel, Generic[T], Tagged): return df - def to_lists(self, fields: Optional[List[str]] = None) -> Dict[str, List]: + def to_dict(self) -> Dict[str, List]: """Converts results field to list of values. Returns @@ -61,11 +61,8 @@ class CommandOutput(GenericModel, Generic[T], Tagged): Dictionary of lists. """ df = self.to_dataframe() - target_fields = df.columns if fields is None else fields results = {} - for field in target_fields: - if field not in df.columns: - choices = ", ".join(df.columns) - raise ValueError(f"Field {field} not found, choose from: {choices}.") + for field in df.columns: results[field] = df[field].tolist() + return results