diff --git a/openbb_platform/obbject_extensions/charting/openbb_charting/core/plotly_ta/data_classes.py b/openbb_platform/obbject_extensions/charting/openbb_charting/core/plotly_ta/data_classes.py index b6588ae1cee..1d47f5b99c1 100644 --- a/openbb_platform/obbject_extensions/charting/openbb_charting/core/plotly_ta/data_classes.py +++ b/openbb_platform/obbject_extensions/charting/openbb_charting/core/plotly_ta/data_classes.py @@ -83,7 +83,7 @@ class TAIndicator: return output def get_argument_values(self, label: str) -> Union[List[Any], Any]: - """Returns arguments values by label.""" + """Return arguments values by label.""" output = [] options = self.get_args(label) if options is not None: @@ -106,7 +106,7 @@ class ChartIndicators: ) def get_indicator(self, name: str) -> Union[TAIndicator, None]: - """Returns indicator with given name.""" + """Return indicator with given name.""" output = None for indicator in self.indicators: # type: ignore if indicator.name == name: @@ -114,7 +114,7 @@ class ChartIndicators: return output def get_indicator_args(self, name: str, label: str) -> Union[Arguments, None]: - """Returns argument values for given indicator and label.""" + """Return argument values for given indicator and label.""" output = None indicator = self.get_indicator(name) if indicator is not None: @@ -131,18 +131,18 @@ class ChartIndicators: """Return dictionary of active indicators and their arguments.""" output = {} if self.indicators: - output = {indicator.name: indicator for indicator in self.indicators} + output = {str(indicator.name): indicator for indicator in self.indicators} return output def get_active_ids(self) -> List[str]: - """Returns list of names of active indicators.""" + """Return list of names of active indicators.""" active_ids = [] if self.indicators: - active_ids = [indicator.name for indicator in self.indicators] + active_ids = [str(indicator.name) for indicator in self.indicators] return active_ids def get_arg_names(self, name: str) -> List[str]: - """Returns list of argument labels for given indicator.""" + """Return list of argument labels for given indicator.""" output = [] indicator = self.get_indicator(name) if indicator is not None: @@ -151,7 +151,7 @@ class ChartIndicators: return output def get_options_dict(self, name: str) -> Dict[str, Optional[Arguments]]: - """Returns dictionary of argument labels and values for given indicator.""" + """Return dictionary of argument labels and values for given indicator.""" output = None options = self.get_arg_names(name) if options: @@ -163,7 +163,7 @@ class ChartIndicators: @staticmethod def get_available_indicators() -> Tuple[str, ...]: - """Returns tuple of available indicators.""" + """Return tuple of available indicators.""" return TAIndicator.__annotations__["name"].__args__ # pylint: disable=E1101 @classmethod @@ -279,7 +279,7 @@ class TA_Data: def get_indicator_data(self, indicator: TAIndicator, **args) -> pd.DataFrame: """ - Returns dataframe with indicator data. + Return dataframe with indicator data. Parameters ---------- @@ -288,7 +288,7 @@ class TA_Data: args : dict Arguments for given indicator - Returns + Return ------- pd.DataFrame Dataframe with indicator data @@ -341,7 +341,7 @@ class TA_Data: return output def to_dataframe(self) -> pd.DataFrame: - """Returns dataframe with all indicators.""" + """Return dataframe with all indicators.""" active_indicators = self.indicators.get_indicators() if not active_indicators: