diff --git a/openbb_sdk/providers/fmp/openbb_fmp/models/balance_sheet.py b/openbb_sdk/providers/fmp/openbb_fmp/models/balance_sheet.py index 649bbdbca5b..93787295abe 100644 --- a/openbb_sdk/providers/fmp/openbb_fmp/models/balance_sheet.py +++ b/openbb_sdk/providers/fmp/openbb_fmp/models/balance_sheet.py @@ -5,7 +5,7 @@ from datetime import ( date as dateType, datetime, ) -from typing import Dict, List, Literal, Optional +from typing import Any, Dict, List, Literal, Optional from openbb_provider.abstract.data import Data from openbb_provider.abstract.fetcher import Fetcher @@ -118,9 +118,7 @@ class FMPBalanceSheetFetcher( ] ): @staticmethod - def transform_query( - query: BalanceSheetQueryParams, extra_params: Optional[Dict] = None - ) -> FMPBalanceSheetQueryParams: + def transform_query(params: Dict[str, Any]) -> FMPBalanceSheetQueryParams: period = "annual" if query.period == "annually" else "quarter" return FMPBalanceSheetQueryParams( symbol=query.symbol, period=period, **extra_params if extra_params else {} # type: ignore @@ -141,4 +139,4 @@ class FMPBalanceSheetFetcher( def transform_data( data: List[FMPBalanceSheetData], ) -> List[BalanceSheetData]: - return data_transformer(data, BalanceSheetData) + return data diff --git a/openbb_sdk/providers/fmp/openbb_fmp/models/cash_flow.py b/openbb_sdk/providers/fmp/openbb_fmp/models/cash_flow.py index 2330d07f061..9a8500e630e 100644 --- a/openbb_sdk/providers/fmp/openbb_fmp/models/cash_flow.py +++ b/openbb_sdk/providers/fmp/openbb_fmp/models/cash_flow.py @@ -5,7 +5,7 @@ from datetime import ( date as dateType, datetime, ) -from typing import Dict, List, Literal, Optional +from typing import Any, Dict, List, Literal, Optional from openbb_provider.abstract.data import Data from openbb_provider.abstract.fetcher import Fetcher @@ -107,9 +107,7 @@ class FMPCashFlowStatementFetcher( ] ): @staticmethod - def transform_query( - query: CashFlowStatementQueryParams, extra_params: Optional[Dict] = None - ) -> FMPCashFlowStatementQueryParams: + def transform_query(params: Dict[str, Any]) -> FMPCashFlowStatementQueryParams: period = "annual" if query.period == "annually" else "quarter" return FMPCashFlowStatementQueryParams( symbol=query.symbol, period=period, **extra_params if extra_params else {} # type: ignore @@ -129,5 +127,5 @@ class FMPCashFlowStatementFetcher( @staticmethod def transform_data( data: List[FMPCashFlowStatementData], - ) -> List[CashFlowStatementData]: - return data_transformer(data, CashFlowStatementData) + ) -> List[FMPCashFlowStatementData]: + return data