fix: datetime.data in series (#20618)

This commit is contained in:
Yongjie Zhao
2022-07-07 09:33:44 +08:00
committed by GitHub
parent c992ff3be4
commit 414cc99ca2
3 changed files with 68 additions and 7 deletions

View File

@@ -32,7 +32,7 @@ from superset.charts.dao import ChartDAO
from superset.common.chart_data import ChartDataResultFormat
from superset.common.db_query_status import QueryStatus
from superset.common.query_actions import get_query_results
from superset.common.utils import dataframe_utils as df_utils
from superset.common.utils import dataframe_utils
from superset.common.utils.query_cache_manager import QueryCacheManager
from superset.connectors.base.models import BaseDatasource
from superset.constants import CacheRegion
@@ -231,7 +231,7 @@ class QueryContextProcessor:
)
if self.enforce_numerical_metrics:
df_utils.df_metrics_to_num(df, query_object)
dataframe_utils.df_metrics_to_num(df, query_object)
df.replace([np.inf, -np.inf], np.nan, inplace=True)
@@ -322,9 +322,7 @@ class QueryContextProcessor:
# multi-dimensional charts
granularity = query_object.granularity
index = granularity if granularity in df.columns else DTTM_ALIAS
if not pd.api.types.is_datetime64_any_dtype(
offset_metrics_df.get(index)
):
if not dataframe_utils.is_datetime_series(offset_metrics_df.get(index)):
raise QueryObjectValidationError(
_(
"A time column must be specified "
@@ -337,7 +335,7 @@ class QueryContextProcessor:
)
# df left join `offset_metrics_df`
offset_df = df_utils.left_join_df(
offset_df = dataframe_utils.left_join_df(
left_df=df,
right_df=offset_metrics_df,
join_keys=join_keys,