From 37bb543dda3ee6520af2a9eefd20b4e7016569bc Mon Sep 17 00:00:00 2001 From: Diogo Sousa Date: Wed, 5 Jul 2023 13:21:06 +0100 Subject: [PATCH] rename addional_types var --- .../openbb_sdk_core/app/static/package_builder.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/openbb_sdk/openbb_sdk_core/openbb_sdk_core/app/static/package_builder.py b/openbb_sdk/openbb_sdk_core/openbb_sdk_core/app/static/package_builder.py index 54e0714621a..68659b0a753 100644 --- a/openbb_sdk/openbb_sdk_core/openbb_sdk_core/app/static/package_builder.py +++ b/openbb_sdk/openbb_sdk_core/openbb_sdk_core/app/static/package_builder.py @@ -251,7 +251,7 @@ class MethodDefinition: @staticmethod def build_func_params(parameter_map: Dict[str, Parameter]) -> str: - ADDITIONAL_TYPES = { + TYPE_EXPANSION = { "data": pd.DataFrame, "start_date": str, "end_date": str, @@ -272,7 +272,7 @@ class MethodDefinition: type_ = MethodDefinition.get_type(field) default = MethodDefinition.get_default(field) - new_type = ADDITIONAL_TYPES.get(name, None) + new_type = TYPE_EXPANSION.get(name, None) updated_type = type_ if new_type is None else Union[type_, new_type] formatted[name] = Parameter( @@ -282,7 +282,7 @@ class MethodDefinition: default=default, ) else: - new_type = ADDITIONAL_TYPES.get(name, None) + new_type = TYPE_EXPANSION.get(name, None) updated_type = ( param.annotation if new_type is None