Skip to content

Commit 564da85

Browse files
authored
[ML] Adjust wire serialization code after backport to 7.x (#63062)
1 parent 0219ac3 commit 564da85

File tree

2 files changed

+14
-30
lines changed

2 files changed

+14
-30
lines changed

x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/ml/dataframe/evaluation/classification/AbstractAucRoc.java

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
*/
66
package org.elasticsearch.xpack.core.ml.dataframe.evaluation.classification;
77

8-
import org.elasticsearch.Version;
98
import org.elasticsearch.common.ParseField;
109
import org.elasticsearch.common.Strings;
1110
import org.elasticsearch.common.io.stream.StreamInput;
@@ -246,11 +245,7 @@ public Result(double score, Long docCount, List<AucRocPoint> curve) {
246245

247246
public Result(StreamInput in) throws IOException {
248247
this.score = in.readDouble();
249-
if (in.getVersion().onOrAfter(Version.V_8_0_0)) {
250-
this.docCount = in.readOptionalLong();
251-
} else {
252-
this.docCount = null;
253-
}
248+
this.docCount = in.readOptionalLong();
254249
this.curve = in.readList(AucRocPoint::new);
255250
}
256251

@@ -279,9 +274,7 @@ public String getMetricName() {
279274
@Override
280275
public void writeTo(StreamOutput out) throws IOException {
281276
out.writeDouble(score);
282-
if (out.getVersion().onOrAfter(Version.V_8_0_0)) {
283-
out.writeOptionalLong(docCount);
284-
}
277+
out.writeOptionalLong(docCount);
285278
out.writeList(curve);
286279
}
287280

x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/ml/dataframe/evaluation/classification/Classification.java

Lines changed: 12 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
*/
66
package org.elasticsearch.xpack.core.ml.dataframe.evaluation.classification;
77

8-
import org.elasticsearch.Version;
98
import org.elasticsearch.common.Nullable;
109
import org.elasticsearch.common.ParseField;
1110
import org.elasticsearch.common.io.stream.StreamInput;
@@ -97,18 +96,14 @@ private static List<EvaluationMetric> defaultMetrics() {
9796
}
9897

9998
public Classification(StreamInput in) throws IOException {
100-
if (in.getVersion().onOrAfter(Version.V_8_0_0)) {
101-
this.fields =
102-
new EvaluationFields(
103-
in.readString(),
104-
in.readOptionalString(),
105-
in.readOptionalString(),
106-
in.readOptionalString(),
107-
in.readOptionalString(),
108-
true);
109-
} else {
110-
this.fields = new EvaluationFields(in.readString(), in.readString(), null, null, null, true);
111-
}
99+
this.fields =
100+
new EvaluationFields(
101+
in.readString(),
102+
in.readOptionalString(),
103+
in.readOptionalString(),
104+
in.readOptionalString(),
105+
in.readOptionalString(),
106+
true);
112107
this.metrics = in.readNamedWriteableList(EvaluationMetric.class);
113108
}
114109

@@ -135,14 +130,10 @@ public String getWriteableName() {
135130
@Override
136131
public void writeTo(StreamOutput out) throws IOException {
137132
out.writeString(fields.getActualField());
138-
if (out.getVersion().onOrAfter(Version.V_8_0_0)) {
139-
out.writeOptionalString(fields.getPredictedField());
140-
out.writeOptionalString(fields.getTopClassesField());
141-
out.writeOptionalString(fields.getPredictedClassField());
142-
out.writeOptionalString(fields.getPredictedProbabilityField());
143-
} else {
144-
out.writeString(fields.getPredictedField());
145-
}
133+
out.writeOptionalString(fields.getPredictedField());
134+
out.writeOptionalString(fields.getTopClassesField());
135+
out.writeOptionalString(fields.getPredictedClassField());
136+
out.writeOptionalString(fields.getPredictedProbabilityField());
146137
out.writeNamedWriteableList(metrics);
147138
}
148139

0 commit comments

Comments
 (0)