From 56727f9560a055c811adc6c3f18ee4d4e743cb9f Mon Sep 17 00:00:00 2001 From: Danglewood <85772166+deeleeramone@users.noreply.github.com> Date: Tue, 2 Apr 2024 22:46:05 -0700 Subject: drop 'provider' in to_df --- openbb_platform/core/openbb_core/app/model/obbject.py | 3 +++ openbb_platform/core/openbb_core/app/utils.py | 3 +++ 2 files changed, 6 insertions(+) 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 -- cgit v1.2.3