summaryrefslogtreecommitdiffstats
path: root/openbb_platform/providers/sec/openbb_sec/models/institutions_search.py
diff options
context:
space:
mode:
Diffstat (limited to 'openbb_platform/providers/sec/openbb_sec/models/institutions_search.py')
-rw-r--r--openbb_platform/providers/sec/openbb_sec/models/institutions_search.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/openbb_platform/providers/sec/openbb_sec/models/institutions_search.py b/openbb_platform/providers/sec/openbb_sec/models/institutions_search.py
index 15815ef6f4a..23a09ec2ba3 100644
--- a/openbb_platform/providers/sec/openbb_sec/models/institutions_search.py
+++ b/openbb_platform/providers/sec/openbb_sec/models/institutions_search.py
@@ -1,5 +1,7 @@
"""SEC Institutions Search Model."""
+# pylint: disable=unused-argument
+
from typing import Any, Dict, List, Optional, Union
from openbb_core.provider.abstract.data import Data
@@ -40,19 +42,17 @@ class SecInstitutionsSearchFetcher(
"""Transform the query."""
return SecInstitutionsSearchQueryParams(**params)
- # pylint: disable=unused-argument
@staticmethod
- def extract_data(
+ async def aextract_data(
query: SecInstitutionsSearchQueryParams,
credentials: Optional[Dict[str, str]],
**kwargs: Any,
) -> List[Dict]:
"""Return the raw data from the SEC endpoint."""
- institutions = get_all_ciks(use_cache=query.use_cache)
+ institutions = await get_all_ciks(use_cache=query.use_cache)
hp = institutions["Institution"].str.contains(query.query, case=False)
return institutions[hp].astype(str).to_dict("records")
- # pylint: disable=unused-argument
@staticmethod
def transform_data(
query: SecInstitutionsSearchQueryParams, data: List[Dict], **kwargs: Any