[ML] Fix serialisation of text embedding updates (elastic#85863)

davidkyle · Apr 13, 2022 · b4f0057 · b4f0057
1 parent 6ca9a15
commit b4f0057
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/docs/changelog/85863.yaml b/docs/changelog/85863.yaml
@@ -0,0 +1,5 @@
+pr: 85863
+summary: Fix serialisation of text embedding updates
+area: Machine Learning
+type: bug
+issues: []
diff --git a/...main/java/org/elasticsearch/xpack/core/ml/inference/MlInferenceNamedXContentProvider.java b/...main/java/org/elasticsearch/xpack/core/ml/inference/MlInferenceNamedXContentProvider.java
@@ -600,11 +600,7 @@ public List<NamedWriteableRegistry.Entry> getNamedWriteables() {
             )
         );
         namedWriteables.add(
-            new NamedWriteableRegistry.Entry(
-                InferenceConfigUpdate.class,
-                TextEmbeddingConfigUpdate.NAME,
-                TextClassificationConfigUpdate::new
-            )
+            new NamedWriteableRegistry.Entry(InferenceConfigUpdate.class, TextEmbeddingConfigUpdate.NAME, TextEmbeddingConfigUpdate::new)
         );
         namedWriteables.add(
             new NamedWriteableRegistry.Entry(