From 22e27ee6975a2236c96e77d161470f1807ff8ccb Mon Sep 17 00:00:00 2001 From: James Chao Date: Tue, 6 Feb 2024 00:41:35 -0500 Subject: [PATCH] fix forecast flattening prediction list error --- raiwidgets/raiwidgets/responsibleai_dashboard_input.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/raiwidgets/raiwidgets/responsibleai_dashboard_input.py b/raiwidgets/raiwidgets/responsibleai_dashboard_input.py index 6c6738da4b..7efd7d4848 100644 --- a/raiwidgets/raiwidgets/responsibleai_dashboard_input.py +++ b/raiwidgets/raiwidgets/responsibleai_dashboard_input.py @@ -445,9 +445,6 @@ def change(x): predictions = convert_to_list( self._analysis.model.forecast(filtered_data_df), EXP_VIZ_ERR_MSG) - # forecast should return a flat list of predictions - if all([len(p) == 1 for p in predictions]): - predictions = [p[0] for p in predictions] return { WidgetRequestResponseConstants.data: predictions }