diff options
Diffstat (limited to 'openbb_platform/core/openbb_core/app/utils.py')
-rw-r--r-- | openbb_platform/core/openbb_core/app/utils.py | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/openbb_platform/core/openbb_core/app/utils.py b/openbb_platform/core/openbb_core/app/utils.py index 4004e3060a9..452300c4334 100644 --- a/openbb_platform/core/openbb_core/app/utils.py +++ b/openbb_platform/core/openbb_core/app/utils.py @@ -3,7 +3,7 @@ import ast import json from datetime import time -from typing import Dict, Iterable, List, Optional, Union +from typing import Dict, List, Optional, Union import numpy as np import pandas as pd @@ -17,7 +17,7 @@ from openbb_core.provider.abstract.data import Data def basemodel_to_df( data: Union[List[Data], Data], - index: Optional[Union[None, str, Iterable]] = None, + index: Optional[str] = None, ) -> pd.DataFrame: """Convert list of BaseModel to a Pandas DataFrame.""" if isinstance(data, list): @@ -44,9 +44,7 @@ def basemodel_to_df( df.set_index("date", inplace=True) df.sort_index(axis=0, inplace=True) else: - df = ( - df.set_index(index) if index is not None and index in df.columns else df - ) + df = df.set_index(index) if index and index in df.columns else df return df @@ -168,6 +166,6 @@ def check_single_item( value: Optional[str], message: Optional[str] = None ) -> Optional[str]: """Check that string contains a single item.""" - if value and ("," in value or ";" in value): + if value and isinstance(value, str) and ("," in value or ";" in value): raise OpenBBError(message if message else "multiple items not allowed") return value |