summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorDanglewood <85772166+deeleeramone@users.noreply.github.com>2024-05-17 17:18:23 -0700
committerDanglewood <85772166+deeleeramone@users.noreply.github.com>2024-05-17 17:18:23 -0700
commitea20273e14f9a704018c7bbb9404517759b1783a (patch)
treebc7e40674f06a44483bfa141526cf0c0a1c98703
parenta3757146b0a307f7e9500a9cc4e89181eca6ed86 (diff)
pylint disable unused-argument
-rw-r--r--openbb_platform/providers/fmp/openbb_fmp/models/equity_screener.py9
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)