diff options
author | Danglewood <85772166+deeleeramone@users.noreply.github.com> | 2024-04-02 22:46:05 -0700 |
---|---|---|
committer | Danglewood <85772166+deeleeramone@users.noreply.github.com> | 2024-04-02 22:46:05 -0700 |
commit | 56727f9560a055c811adc6c3f18ee4d4e743cb9f (patch) | |
tree | 05575e788276dec5fa6c9919b8b0b0524e8dcf0a | |
parent | 83d03bc2544a412d123ec9137e98dbb5d9d2dc7f (diff) |
drop 'provider' in to_dfbugfix/drop-provider-field
-rw-r--r-- | openbb_platform/core/openbb_core/app/model/obbject.py | 3 | ||||
-rw-r--r-- | openbb_platform/core/openbb_core/app/utils.py | 3 |
2 files changed, 6 insertions, 0 deletions
diff --git a/openbb_platform/core/openbb_core/app/model/obbject.py b/openbb_platform/core/openbb_core/app/model/obbject.py index 89232702405..1bc5e14c20f 100644 --- a/openbb_platform/core/openbb_core/app/model/obbject.py +++ b/openbb_platform/core/openbb_core/app/model/obbject.py @@ -226,6 +226,9 @@ class OBBject(Tagged, Generic[T]): except Exception as ex: raise OpenBBError(f"An unexpected error occurred: {ex}") from ex + if "provider" in df.columns: + df.drop(columns=["provider"], inplace=True) + return df def to_polars(self) -> "PolarsDataFrame": diff --git a/openbb_platform/core/openbb_core/app/utils.py b/openbb_platform/core/openbb_core/app/utils.py index f7c6c1ca7dc..a11722f3a26 100644 --- a/openbb_platform/core/openbb_core/app/utils.py +++ b/openbb_platform/core/openbb_core/app/utils.py @@ -46,6 +46,9 @@ def basemodel_to_df( else: df = df.set_index(index) if index and index in df.columns else df + if "provider" in df.columns: + df.drop(columns=["provider"], inplace=True) + return df |