diff options
author | Danglewood <85772166+deeleeramone@users.noreply.github.com> | 2024-05-17 17:18:23 -0700 |
---|---|---|
committer | Danglewood <85772166+deeleeramone@users.noreply.github.com> | 2024-05-17 17:18:23 -0700 |
commit | ea20273e14f9a704018c7bbb9404517759b1783a (patch) | |
tree | bc7e40674f06a44483bfa141526cf0c0a1c98703 | |
parent | a3757146b0a307f7e9500a9cc4e89181eca6ed86 (diff) |
pylint disable unused-argument
-rw-r--r-- | openbb_platform/providers/fmp/openbb_fmp/models/equity_screener.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/openbb_platform/providers/fmp/openbb_fmp/models/equity_screener.py b/openbb_platform/providers/fmp/openbb_fmp/models/equity_screener.py index 1f4fd343519..9f5f9b33277 100644 --- a/openbb_platform/providers/fmp/openbb_fmp/models/equity_screener.py +++ b/openbb_platform/providers/fmp/openbb_fmp/models/equity_screener.py @@ -1,5 +1,7 @@ """FMP Equity Screener Model.""" +# pylint: disable=unused-argument + from typing import Any, Dict, List, Literal, Optional import pandas as pd @@ -8,6 +10,7 @@ from openbb_core.provider.standard_models.equity_screener import ( EquityScreenerData, EquityScreenerQueryParams, ) +from openbb_core.provider.utils.errors import EmptyDataError from openbb_fmp.utils.definitions import EXCHANGES, SECTORS from openbb_fmp.utils.helpers import create_url, get_data from pydantic import Field @@ -171,16 +174,16 @@ class FMPEquityScreenerFetcher( query=query, exclude=["query", "is_symbol", "industry"], ).replace(" ", "%20") - return await get_data(url, **kwargs) + return await get_data(url, **kwargs) # type: ignore @staticmethod def transform_data( query: FMPEquityScreenerQueryParams, data: List[Dict], **kwargs: Any ) -> List[FMPEquityScreenerData]: """Return the transformed data.""" + if not data: + raise EmptyDataError("The request was returned empty.") results = pd.DataFrame(data) - if len(results) == 0: - return [] if query.industry: results = results[ results["sector"].str.contains(query.industry, case=False) |