mypy

OpenBB-finance · May 15, 2024 · c514112 · c514112
1 parent 20069e1
commit c514112
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 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
@@ -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 @@ def transform_data(
     ) -> 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
@@ -79,7 +79,7 @@ def generate_url(date):
             )
             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)