From 9c367d149cf1c09e8d5361c28628b8024c950a53 Mon Sep 17 00:00:00 2001 From: Diogo Sousa Date: Sat, 29 Jul 2023 14:47:17 +0100 Subject: [PATCH] just return the data --- .../openbb_benzinga/models/stock_news.py | 17 +++++------------ 1 file changed, 5 insertions(+), 12 deletions(-) diff --git a/openbb_sdk/providers/benzinga/openbb_benzinga/models/stock_news.py b/openbb_sdk/providers/benzinga/openbb_benzinga/models/stock_news.py index 8403358c93e..a223d0d49b3 100644 --- a/openbb_sdk/providers/benzinga/openbb_benzinga/models/stock_news.py +++ b/openbb_sdk/providers/benzinga/openbb_benzinga/models/stock_news.py @@ -2,7 +2,7 @@ from datetime import datetime -from typing import Dict, List, Literal, Optional +from typing import Any, Dict, List, Literal, Optional from openbb_provider.abstract.fetcher import Fetcher from openbb_provider.helpers import data_transformer, get_querystring @@ -100,15 +100,8 @@ class BenzingaStockNewsFetcher( ] ): @staticmethod - def transform_query( - query: StockNewsQueryParams, extra_params: Optional[Dict] = None - ) -> BenzingaStockNewsQueryParams: - return BenzingaStockNewsQueryParams( - page=query.page, - symbols=query.symbols, # type: ignore - limit=query.limit, - **extra_params if extra_params else {}, - ) + def transform_query(params: Dict[str, Any]) -> BenzingaStockNewsQueryParams: + return BenzingaStockNewsQueryParams(**params) @staticmethod def extract_data( @@ -128,6 +121,6 @@ class BenzingaStockNewsFetcher( return [BenzingaStockNewsData(**d) for d in data] @staticmethod - def transform_data(data: List[BenzingaStockNewsData]) -> List[StockNewsData]: + def transform_data(data: List[BenzingaStockNewsData]) -> List[BenzingaStockNewsData]: processors = {"image": lambda x: "" if x == [] else x[0].url} - return data_transformer(data, StockNewsData, processors) + return data