mypy ignores

This commit is contained in:
Diogo Sousa
2023-07-31 20:21:15 +01:00
parent 96e36867b0
commit 57b6546249
3 changed files with 6 additions and 6 deletions

View File

@@ -78,9 +78,9 @@ class CommandOutput(GenericModel, Generic[T], Tagged):
df = pd.concat(dict_of_df, axis=1) if concat else dict_of_df
else:
df = basemodel_to_df(res, "date")
df = basemodel_to_df(res, "date") # type: ignore
else:
df = basemodel_to_df(res, "date")
df = basemodel_to_df(res, "date") # type: ignore
except Exception as e:
raise OpenBBError("Failed to convert results to DataFrame.") from e
@@ -95,7 +95,7 @@ class CommandOutput(GenericModel, Generic[T], Tagged):
Dict[str, List]
Dictionary of lists.
"""
df = self.to_dataframe().reset_index()
df = self.to_dataframe().reset_index() # type: ignore
results = {}
for field in df.columns:
results[field] = df[field].tolist()

View File

@@ -184,7 +184,7 @@ class ProviderInterface:
if field.required:
if force_optional:
type_ = Optional[type_]
type_ = Optional[type_] # type: ignore
default = None
else:
default = ...

View File

@@ -1,4 +1,4 @@
from typing import Any, Dict, Optional
from typing import Any, Dict, Optional, Type
from openbb_provider.abstract.fetcher import Fetcher, GenericDataType
from openbb_provider.abstract.provider import Provider
@@ -25,7 +25,7 @@ class QueryExecutor:
)
return self.registry.providers[name]
def get_fetcher(self, provider: Provider, model_name: str) -> Fetcher:
def get_fetcher(self, provider: Provider, model_name: str) -> Type[Fetcher]:
"""Get a fetcher from a provider."""
if model_name not in provider.fetcher_dict:
raise ProviderError(