iop
/
superset
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: logging warning on dataframe (don't use python's warnings) (#18111) 

* fix: logging warning on dataframe (don't use python's warnings)

* lint
This commit is contained in:
Daniel Vaz Gaspar 2022-01-20 16:24:30 +00:00 committed by GitHub
parent a5e805ef2a
commit 5fb8b0a13a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
superset/dataframe.py
View File

@ -16,13 +16,15 @@
# under the License.
""" Superset utilities for pandas.DataFrame.
"""
import warnings
import logging
from typing import Any, Dict, List
import pandas as pd
from superset.utils.core import JS_MAX_INTEGER
logger = logging.getLogger(__name__)
def _convert_big_integers(val: Any) -> Any:
"""
@ -43,10 +45,8 @@ def df_to_records(dframe: pd.DataFrame) -> List[Dict[str, Any]]:
:returns: a list of dictionaries reflecting each single row of the DataFrame
"""
if not dframe.columns.is_unique:
warnings.warn(
"DataFrame columns are not unique, some columns will be omitted.",
UserWarning,
stacklevel=2,
logger.warning(
"DataFrame columns are not unique, some columns will be omitted."
)
columns = dframe.columns
return list(