fix the tests

elastic · Nov 10, 2023 · 2d4dc4c · 2d4dc4c
1 parent 9ed028f
commit 2d4dc4c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/...ugin/core/src/main/java/org/elasticsearch/xpack/core/ml/inference/TrainedModelConfig.java b/...ugin/core/src/main/java/org/elasticsearch/xpack/core/ml/inference/TrainedModelConfig.java
@@ -298,7 +298,7 @@ public TrainedModelConfig(StreamInput in) throws IOException {
         } else {
             platformArchitecture = null;
         }
-        if (in.getTransportVersion().onOrAfter(TransportVersions.ML_PACKAGE_LOADER_PLATFORM_ADDED)) {
+        if (in.getTransportVersion().onOrAfter(TransportVersions.ML_TRAINED_MODEL_PREFIX_STRINGS_ADDED)) {
             prefixStrings = in.readOptionalWriteable(TrainedModelPrefixStrings::new);
         }
     }
@@ -486,7 +486,7 @@ public void writeTo(StreamOutput out) throws IOException {
             out.writeOptionalString(platformArchitecture);
         }
 
-        if (out.getTransportVersion().onOrAfter(TransportVersions.ML_PACKAGE_LOADER_PLATFORM_ADDED)) {
+        if (out.getTransportVersion().onOrAfter(TransportVersions.ML_TRAINED_MODEL_PREFIX_STRINGS_ADDED)) {
             out.writeOptionalWriteable(prefixStrings);
         }
     }

diff --git a/...core/src/test/java/org/elasticsearch/xpack/core/ml/inference/TrainedModelConfigTests.java b/...core/src/test/java/org/elasticsearch/xpack/core/ml/inference/TrainedModelConfigTests.java
@@ -463,6 +463,9 @@ protected TrainedModelConfig mutateInstanceForVersion(TrainedModelConfig instanc
         if (version.before(TransportVersions.ML_TRAINED_MODEL_CONFIG_PLATFORM_ADDED)) {
             builder.setPlatformArchitecture(null);
         }
+        if (version.before(TransportVersions.ML_TRAINED_MODEL_PREFIX_STRINGS_ADDED)) {
+            builder.setPrefixStrings(null);
+        }
         return builder.build();
     }
 }