From 5af25441630a8ba173c89f70c8f745539db4656a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Istv=C3=A1n=20Zolt=C3=A1n=20Szab=C3=B3?=
 <szabosteve@gmail.com>
Date: Wed, 7 Sep 2022 10:23:34 +0200
Subject: [PATCH] [ML] Modifies script for DFA screenshots. (#140105)

---
 .../apps/ml_docs/data_frame_analytics/classification.ts        | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/x-pack/test/screenshot_creation/apps/ml_docs/data_frame_analytics/classification.ts b/x-pack/test/screenshot_creation/apps/ml_docs/data_frame_analytics/classification.ts
index 5fcbd6f2cb998..63f40f46023b8 100644
--- a/x-pack/test/screenshot_creation/apps/ml_docs/data_frame_analytics/classification.ts
+++ b/x-pack/test/screenshot_creation/apps/ml_docs/data_frame_analytics/classification.ts
@@ -126,6 +126,7 @@ export default function ({ getService }: FtrProviderContext) {
       await ml.dataFrameAnalyticsResults.expandClassificationEvaluationSection(false);
       await ml.dataFrameAnalyticsResults.expandFeatureImportanceSection(false);
       await ml.dataFrameAnalyticsResults.expandScatterplotMatrixSection(false);
+      await ml.dataFrameAnalyticsResults.enableResultsTablePreviewHistogramCharts(true);
       await ml.dataFrameAnalyticsResults.scrollAnalysisIntoView();
       await commonScreenshots.removeFocusFromElement();
       await commonScreenshots.takeScreenshot(
@@ -156,7 +157,7 @@ export default function ({ getService }: FtrProviderContext) {
         'confusion-matrix-binary-accuracy',
         screenshotDirectories
       );
-      await ml.dataFrameAnalyticsResults.scrollRocCurveChartIntoView();
+      await ml.dataFrameAnalyticsResults.scrollClassificationEvaluationIntoView();
       await commonScreenshots.takeScreenshot(
         'flights-classification-roc-curve',
         screenshotDirectories