Fix failing test.

afoucret · afoucret · commit 922499bf6f77 · 2025-11-13T16:38:50.000+01:00
diff --git a/x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/elastic/authorization/AuthorizationPollerTests.java b/x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/elastic/authorization/AuthorizationPollerTests.java
@@ -229,8 +229,7 @@ public void testDoesNotAttemptToStoreModelIds_ThatHaveATaskTypeThatTheEISIntegra
                         List.of(
                             new ElasticInferenceServiceAuthorizationResponseEntity.AuthorizedModel(
                                 InternalPreconfiguredEndpoints.DEFAULT_ELSER_2_MODEL_ID,
-                                // EIS does not yet support completions so this model will be ignored
-                                EnumSet.of(TaskType.COMPLETION)
+                                EnumSet.noneOf(TaskType.class)
                             )
                         )
                     )

Original file line number	Diff line number	Diff line change
`@@ -229,8 +229,7 @@ public void testDoesNotAttemptToStoreModelIds_ThatHaveATaskTypeThatTheEISIntegra`
`229`	`229`	`List.of(`
`230`	`230`	`new ElasticInferenceServiceAuthorizationResponseEntity.AuthorizedModel(`
`231`	`231`	`InternalPreconfiguredEndpoints.DEFAULT_ELSER_2_MODEL_ID,`
`232`		`- // EIS does not yet support completions so this model will be ignored`
`233`		`- EnumSet.of(TaskType.COMPLETION)`
	`232`	`+ EnumSet.noneOf(TaskType.class)`
`234`	`233`	`)`
`235`	`234`	`)`
`236`	`235`	`)`