From d8c4fa13d9f20fd56eb8af59e66cb6676e388024 Mon Sep 17 00:00:00 2001 From: David Kyle Date: Mon, 20 Oct 2025 16:57:31 +0100 Subject: [PATCH] [ML] Fix casting in RegressionIT test (#136743) (cherry picked from commit d8854e5885a67254361538202c0b8576aa35deab) # Conflicts: # muted-tests.yml --- muted-tests.yml | 3 --- .../org/elasticsearch/xpack/ml/integration/RegressionIT.java | 4 ++-- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/muted-tests.yml b/muted-tests.yml index 9c048e5041178..b10127de73d82 100644 --- a/muted-tests.yml +++ b/muted-tests.yml @@ -597,9 +597,6 @@ tests: - class: org.elasticsearch.datastreams.DataStreamIndexSettingsProviderTests method: testGetAdditionalIndexSettings issue: https://github.com/elastic/elasticsearch/issues/135972 -- class: org.elasticsearch.xpack.ml.integration.RegressionIT - method: testAliasFields - issue: https://github.com/elastic/elasticsearch/issues/137377 - class: org.elasticsearch.upgrades.SearchStatesIT method: testCanMatch issue: https://github.com/elastic/elasticsearch/issues/137687 diff --git a/x-pack/plugin/ml/qa/native-multi-node-tests/src/javaRestTest/java/org/elasticsearch/xpack/ml/integration/RegressionIT.java b/x-pack/plugin/ml/qa/native-multi-node-tests/src/javaRestTest/java/org/elasticsearch/xpack/ml/integration/RegressionIT.java index a06b85131814d..10fc4721cd4eb 100644 --- a/x-pack/plugin/ml/qa/native-multi-node-tests/src/javaRestTest/java/org/elasticsearch/xpack/ml/integration/RegressionIT.java +++ b/x-pack/plugin/ml/qa/native-multi-node-tests/src/javaRestTest/java/org/elasticsearch/xpack/ml/integration/RegressionIT.java @@ -630,8 +630,8 @@ public void testAliasFields() throws Exception { assertThat(resultsObject.containsKey(predictionField), is(true)); assertThat(resultsObject.containsKey("is_training"), is(true)); - int featureValue = (int) destDoc.get("field_1"); - double predictionValue = (double) resultsObject.get(predictionField); + int featureValue = ((Number) destDoc.get("field_1")).intValue(); + double predictionValue = ((Number) resultsObject.get(predictionField)).doubleValue(); predictionErrorSum += Math.abs(predictionValue - 2 * featureValue); } // We assert on the mean prediction error in order to reduce the probability