fix forecast flattening prediction list error

microsoft · Feb 6, 2024 · 477a7d1 · 477a7d1
1 parent 5eed28f
commit 477a7d1
Showing 1 changed file with 0 additions and 3 deletions.
diff --git 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
             }