summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorDanglewood <85772166+deeleeramone@users.noreply.github.com>2024-05-15 00:43:02 -0700
committerDanglewood <85772166+deeleeramone@users.noreply.github.com>2024-05-15 00:43:02 -0700
commitc51411227b33196d9a212e91009e456c9482a6ed (patch)
tree4735851553dfb29d4f379b856a9ce88b193c3bd0
parent20069e17887910b5e9c4b138d43a5c89ff555420 (diff)
mypy
-rw-r--r--openbb_platform/providers/econdb/openbb_econdb/models/yield_curve.py4
-rw-r--r--openbb_platform/providers/fmp/openbb_fmp/models/yield_curve.py2
2 files changed, 3 insertions, 3 deletions
diff --git a/openbb_platform/providers/econdb/openbb_econdb/models/yield_curve.py b/openbb_platform/providers/econdb/openbb_econdb/models/yield_curve.py
index ec8093b26ae..e6de41b3a5b 100644
--- a/openbb_platform/providers/econdb/openbb_econdb/models/yield_curve.py
+++ b/openbb_platform/providers/econdb/openbb_econdb/models/yield_curve.py
@@ -25,7 +25,7 @@ class EconDbYieldCurveQueryParams(YieldCurveQueryParams):
__json_schema_extra__ = {
"date": ["multiple_items_allowed"],
}
- country: Literal[COUNTRIES] = Field(
+ country: COUNTRIES = Field(
default="united_states",
description=QUERY_DESCRIPTIONS.get("country", ""),
json_schema_extra={"choices": DAILY_COUNTRIES},
@@ -118,7 +118,7 @@ class EconDbYieldCurveFetcher(
) -> AnnotatedResult[List[EconDbYieldCurveData]]:
"""Transform the data."""
maturity_order = list(DAILY[query.country].values())
- dates = query.date.split(",")
+ dates = query.date.split(",") # type: ignore
dates_list = DatetimeIndex(dates)
new_data: Dict = {}
metadata: Dict = {}
diff --git a/openbb_platform/providers/fmp/openbb_fmp/models/yield_curve.py b/openbb_platform/providers/fmp/openbb_fmp/models/yield_curve.py
index 634795023b2..2360a8d0233 100644
--- a/openbb_platform/providers/fmp/openbb_fmp/models/yield_curve.py
+++ b/openbb_platform/providers/fmp/openbb_fmp/models/yield_curve.py
@@ -79,7 +79,7 @@ class FMPYieldCurveFetcher(
)
return url
- dates = query.date.split(",")
+ dates = query.date.split(",") # type: ignore
urls = [generate_url(date) for date in dates]
return await amake_requests(urls, **kwargs)