openbb-charting 2.1.3__tar.gz → 2.2.0__tar.gz

{openbb_charting-2.1.3 → openbb_charting-2.2.0}/PKG-INFO

@@ -1,19 +1,20 @@
 Metadata-Version: 2.1
 Name: openbb-charting
-Version: 2.1.3
+Version: 2.2.0
 Summary: Charting extension for OpenBB
 License: AGPL-3.0-only
 Author: OpenBB Team
 Author-email: hello@openbb.co
-Requires-Python: >=3.8,<3.12
+Requires-Python: >=3.9,<3.13
 Classifier: License :: OSI Approved :: GNU Affero General Public License v3
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: nbformat (>=5.9.2,<6.0.0)
-Requires-Dist: openbb-core (>=1.2.4,<2.0.0)
+Requires-Dist: numpy (>=1,<2)
+Requires-Dist: openbb-core (>=1.3.0,<2.0.0)
 Requires-Dist: pandas-ta (>=0.3.14b,<0.4.0)
 Requires-Dist: plotly (>=5.17.0,<6.0.0)
 Requires-Dist: pywry (>=0.6.1,<0.7.0)

openbb_charting-2.2.0/openbb_charting/__init__.py

@@ -0,0 +1,25 @@
+"""OpenBB OBBject extension for charting."""
+
+import warnings
+
+from openbb_core.app.model.extension import Extension
+
+warnings.filterwarnings(
+    "ignore",
+    category=UserWarning,
+    module="openbb_core.app.model.extension",
+)
+
+
+def get_charting_module():
+    """Get the Charting module."""
+    # pylint: disable=import-outside-toplevel
+    import importlib
+
+    _Charting = importlib.import_module("openbb_charting.charting").Charting
+    return _Charting
+
+
+ext = Extension(name="charting", description="Create custom charts from OBBject data.")
+
+Charting = ext.obbject_accessor(get_charting_module())

{openbb_charting-2.1.3 → openbb_charting-2.2.0}/openbb_charting/charting.py

@@ -3,6 +3,7 @@
 # pylint: disable=too-many-arguments,unused-argument
 
 from typing import (
+    TYPE_CHECKING,
     Any,
     Callable,
     ClassVar,
@@ -16,23 +17,23 @@ from typing import (
 )
 from warnings import warn
 
-import numpy as np
-import pandas as pd
 from importlib_metadata import entry_points
 from openbb_core.app.model.charts.chart import Chart
 from openbb_core.app.model.obbject import OBBject
-from openbb_core.app.utils import basemodel_to_df, convert_to_basemodel
 from openbb_core.provider.abstract.data import Data
-from plotly.graph_objs import Figure
 
-from openbb_charting.charts.generic_charts import bar_chart, line_chart
 from openbb_charting.charts.helpers import (
     get_charting_functions,
     get_charting_functions_list,
 )
-from openbb_charting.core.backend import Backend, create_backend, get_backend
-from openbb_charting.core.openbb_figure import OpenBBFigure
-from openbb_charting.query_params import ChartParams, IndicatorsParams
+
+if TYPE_CHECKING:
+    from numpy import ndarray  # noqa
+    from pandas import DataFrame, Series  # noqa
+    from plotly.graph_objs import Figure  # noqa
+    from openbb_charting.core.openbb_figure import OpenBBFigure  # noqa
+    from openbb_charting.query_params import ChartParams  # noqa
+    from openbb_charting.core.backend import Backend  # noqa
 
 
 class Charting:
@@ -64,18 +65,24 @@ class Charting:
         entry_point.load()
         for entry_point in entry_points(group="openbb_charting_extension")
     ]
+    _format = "plotly"  # the charts computed by this extension will be in plotly format
 
     def __init__(self, obbject):
         """Initialize Charting extension."""
         # pylint: disable=import-outside-toplevel
-        from openbb_core.app.model.charts.charting_settings import ChartingSettings
+        import importlib  # noqa
+
+        charting_settings_module = importlib.import_module(
+            "openbb_core.app.model.charts.charting_settings", "ChartingSettings"
+        )
+        ChartingSettings = charting_settings_module.ChartingSettings
 
         self._obbject: OBBject = obbject
         self._charting_settings = ChartingSettings(
             user_settings=self._obbject._user_settings,  # type: ignore
             system_settings=self._obbject._system_settings,  # type: ignore
         )
-        self._backend: Backend = self._handle_backend()
+        self._backend = self._handle_backend()
         self._functions: Dict[str, Callable] = self._get_functions()
 
     @classmethod
@@ -84,6 +91,9 @@ class Charting:
 
         Without assigning to a variable, it will print the the information to the console.
         """
+        # pylint: disable=import-outside-toplevel
+        from openbb_charting.query_params import IndicatorsParams
+
         return IndicatorsParams()
 
     @classmethod
@@ -103,8 +113,11 @@ class Charting:
 
         return functions
 
-    def _handle_backend(self) -> Backend:
+    def _handle_backend(self) -> "Backend":
         """Create and start the backend."""
+        # pylint: disable=import-outside-toplevel
+        from openbb_charting.core.backend import create_backend, get_backend
+
         create_backend(self._charting_settings)
         backend = get_backend()
         backend.start(debug=self._charting_settings.debug_mode)
@@ -121,11 +134,15 @@ class Charting:
             )
         return self._functions[adjusted_route]
 
-    def get_params(self) -> ChartParams:
+    def get_params(self) -> Union["ChartParams", None]:
         """Return the ChartQueryParams class for the function the OBBject was created from.
 
         Without assigning to a variable, it will print the docstring to the console.
+        If the class is not defined, the help for the function will be returned.
         """
+        # pylint: disable=import-outside-toplevel
+        from openbb_charting.query_params import ChartParams
+
         if self._obbject._route is None:  # pylint: disable=protected-access
             raise ValueError("OBBject was initialized with no function route.")
         charting_function = (
@@ -133,23 +150,32 @@ class Charting:
         ).replace("/", "_")[1:]
         if hasattr(ChartParams, charting_function):
             return getattr(ChartParams, charting_function)()
-        raise ValueError(
-            f"Error: No chart parameters are defined for the route: {charting_function}"
+
+        return help(  # type: ignore
+            self._get_chart_function(  # pylint: disable=protected-access
+                self._obbject.extra[  # pylint: disable=protected-access
+                    "metadata"
+                ].route
+            )
         )
 
     def _prepare_data_as_df(
-        self, data: Optional[Union[pd.DataFrame, pd.Series]]
-    ) -> Tuple[pd.DataFrame, bool]:
+        self, data: Optional[Union["DataFrame", "Series"]]
+    ) -> Tuple["DataFrame", bool]:
         """Convert supplied data to a DataFrame."""
+        # pylint: disable=import-outside-toplevel
+        from openbb_core.app.utils import basemodel_to_df, convert_to_basemodel
+        from pandas import DataFrame, Series
+
         has_data = (
-            isinstance(data, (Data, pd.DataFrame, pd.Series)) and not data.empty  # type: ignore
+            isinstance(data, (Data, DataFrame, Series)) and not data.empty  # type: ignore
         ) or (bool(data))
         index = (
             data.index.name
-            if has_data and isinstance(data, (pd.DataFrame, pd.Series))
+            if has_data and isinstance(data, (DataFrame, Series))
             else None
         )
-        data_as_df: pd.DataFrame = (
+        data_as_df: DataFrame = (
             basemodel_to_df(convert_to_basemodel(data), index=index)
             if has_data
             else self._obbject.to_dataframe(index=index)
@@ -166,11 +192,11 @@ class Charting:
         data: Union[
             list,
             dict,
-            pd.DataFrame,
-            List[pd.DataFrame],
-            pd.Series,
-            List[pd.Series],
-            np.ndarray,
+            "DataFrame",
+            List["DataFrame"],
+            "Series",
+            List["Series"],
+            "ndarray",
             Data,
         ],
         index: Optional[str] = None,
@@ -189,12 +215,12 @@ class Charting:
         same_axis: bool = False,
         render: bool = True,
         **kwargs,
-    ) -> Union[OpenBBFigure, Figure, None]:
+    ) -> Union["OpenBBFigure", "Figure", None]:
         """Create a line chart from external data and render a chart or return the OpenBBFigure.
 
         Parameters
         ----------
-        data : Union[Data, pd.DataFrame, pd.Series]
+        data : Union[Data, DataFrame, Series]
             Data to be plotted (OHLCV data).
         index : Optional[str], optional
             Index column, by default None
@@ -232,6 +258,9 @@ class Charting:
         **kwargs: Dict[str, Any]
             Extra parameters to be passed to `figure.show()`
         """
+        # pylint: disable=import-outside-toplevel
+        from openbb_charting.charts.generic_charts import line_chart
+
         fig = line_chart(
             data=data,
             index=index,
@@ -261,11 +290,11 @@ class Charting:
         data: Union[
             list,
             dict,
-            pd.DataFrame,
-            List[pd.DataFrame],
-            pd.Series,
-            List[pd.Series],
-            np.ndarray,
+            "DataFrame",
+            List["DataFrame"],
+            "Series",
+            List["Series"],
+            "ndarray",
             Data,
         ],
         x: str,
@@ -283,12 +312,12 @@ class Charting:
         bar_kwargs: Optional[Dict[str, Any]] = None,
         render: bool = True,
         **kwargs,
-    ) -> Union[OpenBBFigure, Figure, None]:
+    ) -> Union["OpenBBFigure", "Figure", None]:
         """Create a bar chart on a single x-axis with one or more values for the y-axis.
 
         Parameters
         ----------
-        data : Union[list, dict, pd.DataFrame, List[pd.DataFrame], pd.Series, List[pd.Series], np.ndarray, Data]
+        data : Union[list, dict, DataFrame, List[DataFrame], Series, List[Series], ndarray, Data]
             Data to plot.
         x : str
             The x-axis column name.
@@ -315,6 +344,9 @@ class Charting:
         OpenBBFigure
             The OpenBBFigure object.
         """
+        # pylint: disable=import-outside-toplevel
+        from openbb_charting.charts.generic_charts import bar_chart
+
         fig = bar_chart(
             data=data,
             x=x,
@@ -355,7 +387,7 @@ class Charting:
             fig, content = charting_function(**kwargs)
             fig = self._set_chart_style(fig)
             content = fig.show(external=True, **kwargs).to_plotly_json()
-            self._obbject.chart = Chart(fig=fig, content=content)
+            self._obbject.chart = Chart(fig=fig, content=content, format=self._format)
             if render:
                 fig.show(**kwargs)
         except Exception:  # pylint: disable=W0718
@@ -363,7 +395,9 @@ class Charting:
                 fig = self.create_line_chart(data=self._obbject.results, render=False, **kwargs)  # type: ignore
                 fig = self._set_chart_style(fig)
                 content = fig.show(external=True, **kwargs).to_plotly_json()  # type: ignore
-                self._obbject.chart = Chart(fig=fig, content=content)
+                self._obbject.chart = Chart(
+                    fig=fig, content=content, format=self._format
+                )
                 if render:
                     return fig.show(**kwargs)  # type: ignore
             except Exception as e:
@@ -378,11 +412,11 @@ class Charting:
             Union[
                 list,
                 dict,
-                pd.DataFrame,
-                List[pd.DataFrame],
-                pd.Series,
-                List[pd.Series],
-                np.ndarray,
+                "DataFrame",
+                List["DataFrame"],
+                "Series",
+                List["Series"],
+                "ndarray",
                 Data,
             ]
         ] = None,
@@ -404,7 +438,7 @@ class Charting:
 
         Parameters
         ----------
-        data : Union[Data, pd.DataFrame, pd.Series]
+        data : Union[Data, DataFrame, Series]
             Data to be plotted.
         indicators : Dict[str, Dict[str, Any]], optional
             Indicators to be plotted, by default None
@@ -474,7 +508,9 @@ class Charting:
                 fig = self.create_line_chart(data=data_as_df, render=False, **kwargs)
                 fig = self._set_chart_style(fig)
                 content = fig.show(external=True, **kwargs).to_plotly_json()  # type: ignore
-                self._obbject.chart = Chart(fig=fig, content=content)
+                self._obbject.chart = Chart(
+                    fig=fig, content=content, format=self._format
+                )
                 if render:
                     return fig.show(**kwargs)  # type: ignore
             except Exception as e:  # pylint: disable=W0718
@@ -482,7 +518,7 @@ class Charting:
                     "Failed to automatically create a generic chart with the data provided."
                 ) from e
 
-    def _set_chart_style(self, figure: Figure):
+    def _set_chart_style(self, figure: "Figure"):
         """Set the user preference for light or dark mode."""
         style = self._charting_settings.chart_style
         font_color = "black" if style == "light" else "white"
@@ -512,21 +548,24 @@ class Charting:
 
     def table(
         self,
-        data: Optional[Union[pd.DataFrame, pd.Series]] = None,
+        data: Optional[Union["DataFrame", "Series"]] = None,
         title: str = "",
     ):
         """Display an interactive table.
 
         Parameters
         ----------
-        data : Optional[Union[pd.DataFrame, pd.Series]], optional
+        data : Optional[Union[DataFrame, Series]], optional
             Data to be plotted, by default None.
             If no data is provided the OBBject results will be used.
         title : str, optional
             Title of the table, by default "".
         """
+        # pylint: disable=import-outside-toplevel
+        from pandas import RangeIndex
+
         data_as_df, _ = self._prepare_data_as_df(data)
-        if isinstance(data_as_df.index, pd.RangeIndex):
+        if isinstance(data_as_df.index, RangeIndex):
             data_as_df.reset_index(inplace=True, drop=True)
         else:
             data_as_df.reset_index(inplace=True)
@@ -542,9 +581,7 @@ class Charting:
                 warn(f"Failed to show figure with backend. {e}")
 
         else:
-            from plotly import (  # pylint:disable=import-outside-toplevel
-                optional_imports,
-            )
+            from plotly import optional_imports
 
             ipython_display = optional_imports.get_module("IPython.display")
             if ipython_display:

{openbb_charting-2.1.3 → openbb_charting-2.2.0}/openbb_charting/charts/generic_charts.py

@@ -2,13 +2,10 @@
 
 # pylint: disable=too-many-arguments,unused-argument,too-many-locals, too-many-branches, too-many-lines, too-many-statements, use-dict-literal, broad-exception-caught, too-many-nested-blocks
 
-from typing import Any, Dict, List, Literal, Optional, Union
+from typing import TYPE_CHECKING, Any, Dict, List, Literal, Optional, Union
 
-import numpy as np
-import pandas as pd
 from openbb_core.app.utils import basemodel_to_df, convert_to_basemodel
 from openbb_core.provider.abstract.data import Data
-from plotly.graph_objs import Figure
 
 from openbb_charting.charts.helpers import (
     calculate_returns,
@@ -16,19 +13,24 @@ from openbb_charting.charts.helpers import (
     z_score_standardization,
 )
 from openbb_charting.core.chart_style import ChartStyle
-from openbb_charting.core.openbb_figure import OpenBBFigure
 from openbb_charting.styles.colors import LARGE_CYCLER
 
+if TYPE_CHECKING:
+    from numpy import ndarray  # noqa
+    from pandas import DataFrame, Series  # noqa
+    from plotly.graph_objs import Figure  # noqa
+    from openbb_charting.core.openbb_figure import OpenBBFigure  # noqa
+
 
 def line_chart(  # noqa: PLR0912
     data: Union[
         list,
         dict,
-        pd.DataFrame,
-        List[pd.DataFrame],
-        pd.Series,
-        List[pd.Series],
-        np.ndarray,
+        "DataFrame",
+        List["DataFrame"],
+        "Series",
+        List["Series"],
+        "ndarray",
         Data,
     ],
     index: Optional[str] = None,
@@ -46,31 +48,35 @@ def line_chart(  # noqa: PLR0912
     returns: bool = False,
     same_axis: bool = False,
     **kwargs,
-) -> Union[OpenBBFigure, Figure]:
+) -> Union["OpenBBFigure", "Figure"]:
     """Create a line chart."""
+    # pylint: disable=import-outside-toplevel
+    from pandas import DataFrame, Series, to_datetime  # noqa
+    from openbb_charting.core.openbb_figure import OpenBBFigure  # noqa
+
     if data is None:
         raise ValueError("Error: Data is a required field.")
 
     auto_layout = False
     index = (
         data.index.name
-        if isinstance(data, (pd.DataFrame, pd.Series))
+        if isinstance(data, (DataFrame, Series))
         else index if index is not None else x if x is not None else "date"
     )
-    df: pd.DataFrame = (
-        basemodel_to_df(convert_to_basemodel(data), index=index)
-    ).dropna(how="all", axis=1)
+    df: DataFrame = (basemodel_to_df(convert_to_basemodel(data), index=index)).dropna(
+        how="all", axis=1
+    )
 
     if df.index.name is None:
         if "date" in df.columns:
-            df.date = df.date.apply(pd.to_datetime)
+            df.date = df.date.apply(to_datetime)
             df.set_index("date", inplace=True)
         else:
             found_index = False
             for col in df.columns:
                 if df[col].dtype == "object":
                     try:
-                        df[col] = df[col].apply(pd.to_datetime)
+                        df[col] = df[col].apply(to_datetime)
                         index = df[col].name  # type: ignore
                         df.set_index(col, inplace=True)
                         df.index.name = "date"
@@ -326,11 +332,11 @@ def bar_chart(  # noqa: PLR0912
     data: Union[
         list,
         dict,
-        pd.DataFrame,
-        List[pd.DataFrame],
-        pd.Series,
-        List[pd.Series],
-        np.ndarray,
+        "DataFrame",
+        List["DataFrame"],
+        "Series",
+        List["Series"],
+        "ndarray",
         Data,
     ],
     x: str,
@@ -345,13 +351,13 @@ def bar_chart(  # noqa: PLR0912
     bar_kwargs: Optional[Dict[str, Any]] = None,
     layout_kwargs: Optional[Dict[str, Any]] = None,
     **kwargs,
-) -> Union[OpenBBFigure, Figure]:
+) -> Union["OpenBBFigure", "Figure"]:
     """Create a vertical bar chart on a single x-axis with one or more values for the y-axis.
 
     Parameters
     ----------
     data : Union[
-        list, dict, pd.DataFrame, List[pd.DataFrame], pd.Series, List[pd.Series], np.ndarray, Data
+        list, dict, "DataFrame", List["DataFrame"], "Series", List["Series"], "ndarray", Data
     ]
         Data to plot.
     x : str
@@ -380,6 +386,9 @@ def bar_chart(  # noqa: PLR0912
     OpenBBFigure
         The OpenBBFigure object.
     """
+    # pylint: disable=import-outside-toplevel
+    from openbb_charting.core.openbb_figure import OpenBBFigure
+
     try:
         figure = OpenBBFigure()
     except Exception as _:
@@ -492,7 +501,7 @@ def bar_increasing_decreasing(  # pylint: disable=W0102
     orientation: Literal["h", "v"] = "h",
     barmode: Literal["group", "stack", "relative", "overlay"] = "relative",
     layout_kwargs: Optional[Dict[str, Any]] = None,
-) -> Union[OpenBBFigure, Figure]:
+) -> Union["OpenBBFigure", "Figure"]:
     """Create a bar chart with increasing and decreasing values represented by two colors.
 
     Parameters
@@ -521,6 +530,10 @@ def bar_increasing_decreasing(  # pylint: disable=W0102
     OpenBBFigure
         The OpenBBFigure object.
     """
+    # pylint: disable=import-outside-toplevel
+    from openbb_charting.core.openbb_figure import OpenBBFigure  # noqa
+    from pandas import Series  # noqa
+
     try:
         figure = OpenBBFigure()
     except Exception as _:
@@ -538,7 +551,7 @@ def bar_increasing_decreasing(  # pylint: disable=W0102
     # figure.update_layout(ChartStyle().plotly_template.get("layout", {}))
 
     try:
-        data = pd.Series(data=values, index=keys)
+        data = Series(data=values, index=keys)
         increasing_data = data[data > 0]  # type: ignore
         decreasing_data = data[data < 0]  # type: ignore
     except Exception as e:

{openbb_charting-2.1.3 → openbb_charting-2.2.0}/openbb_charting/charts/helpers.py

@@ -1,14 +1,16 @@
 """Helper functions for charting."""
 
-from inspect import getmembers, getsource, isfunction
-from typing import Callable, Dict, List, Type
+from typing import TYPE_CHECKING, Callable, Dict, List, Type
 
-import pandas as pd
-from pandas_ta import candles
+if TYPE_CHECKING:
+    from pandas import DataFrame, Series
 
 
 def get_charting_functions(view: Type) -> Dict[str, Callable]:
     """Discover charting functions."""
+    # pylint: disable=import-outside-toplevel
+    from inspect import getmembers, getsource, isfunction
+
     implemented_functions: Dict[str, Callable] = {}
 
     for name, obj in getmembers(view, isfunction):
@@ -27,22 +29,25 @@ def get_charting_functions_list(view: Type) -> List[str]:
     return list(get_charting_functions(view).keys())
 
 
-def z_score_standardization(data: pd.Series) -> pd.Series:
+def z_score_standardization(data: "Series") -> "Series":
     """Z-Score Standardization Method."""
     return (data - data.mean()) / data.std()
 
 
-def calculate_returns(data: pd.Series) -> pd.Series:
+def calculate_returns(data: "Series") -> "Series":
     """Calculate the returns of a column."""
     return ((1 + data.pct_change().dropna()).cumprod() - 1) * 100
 
 
 def should_share_axis(
-    df: pd.DataFrame, col1: str, col2: str, threshold: float = 0.15
+    df: "DataFrame", col1: str, col2: str, threshold: float = 0.15
 ) -> bool:
     """Determine whether two columns should share an axis."""
+    # pylint: disable=import-outside-toplevel
+    from pandas import Series
+
     try:
-        if isinstance(df, pd.Series):
+        if isinstance(df, Series):
             df = df.to_frame()
         range1 = df[col1].max() - df[col1].min()
         range2 = df[col2].max() - df[col2].min()
@@ -56,19 +61,22 @@ def should_share_axis(
         return False
 
 
-def heikin_ashi(data: pd.DataFrame) -> pd.DataFrame:
+def heikin_ashi(data: "DataFrame") -> "DataFrame":
     """Return OHLC data as Heikin Ashi Candles.
 
     Parameters
     ----------
-    data: pd.DataFrame
+    data: DataFrame
         DataFrame containing OHLC data.
 
     Returns
     -------
-    pd.DataFrame
+    DataFrame
         DataFrame copy with Heikin Ashi candle calculations.
     """
+    # pylint: disable=import-outside-toplevel
+    from pandas_ta import candles
+
     df = data.copy()
 
     check_columns = ["open", "high", "low", "close"]

{openbb_charting-2.1.3 → openbb_charting-2.2.0}/openbb_charting/charts/price_historical.py

@@ -1,29 +1,34 @@
 """Price historical charting utility."""
 
-from typing import Any, Dict, Tuple
-
-import pandas as pd
-from openbb_core.app.utils import basemodel_to_df
-
-from openbb_charting.charts.helpers import (
-    calculate_returns,
-    heikin_ashi,
-    should_share_axis,
-    z_score_standardization,
-)
-from openbb_charting.core.chart_style import ChartStyle
-from openbb_charting.core.openbb_figure import OpenBBFigure
-from openbb_charting.core.plotly_ta.ta_class import PlotlyTA
+# pylint: disable=too-many-branches, too-many-locals, unused-argument
+
+
+from typing import TYPE_CHECKING, Any, Dict, Tuple
+
 from openbb_charting.styles.colors import LARGE_CYCLER
 
-# pylint: disable=too-many-branches
+if TYPE_CHECKING:
+    from openbb_charting.core.openbb_figure import OpenBBFigure
 
 
 def price_historical(  # noqa: PLR0912
     **kwargs,
-) -> Tuple[OpenBBFigure, Dict[str, Any]]:
+) -> Tuple["OpenBBFigure", Dict[str, Any]]:
     """Equity Price Historical Chart."""
-    if "data" in kwargs and isinstance(kwargs["data"], pd.DataFrame):
+    # pylint: disable=import-outside-toplevel
+    from pandas import DataFrame  # noqa
+    from openbb_core.app.utils import basemodel_to_df  # noqa
+    from openbb_charting.core.openbb_figure import OpenBBFigure  # noqa
+    from openbb_charting.core.plotly_ta.ta_class import PlotlyTA  # noqa
+    from openbb_charting.core.chart_style import ChartStyle  # noqa
+    from openbb_charting.charts.helpers import (  # noqa
+        calculate_returns,
+        heikin_ashi,
+        should_share_axis,
+        z_score_standardization,
+    )
+
+    if "data" in kwargs and isinstance(kwargs["data"], DataFrame):
         data = kwargs["data"]
     elif "data" in kwargs and isinstance(kwargs["data"], list):
         data = basemodel_to_df(kwargs["data"], index=kwargs.get("index", "date"))  # type: ignore