Add missing preventDeletionLock.remove in corner case (elastic#115010)

dan-rubinstein · Oct 17, 2024 · 831d55f · 831d55f
1 parent d1f26ab
commit 831d55f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/...gin/inference/src/main/java/org/elasticsearch/xpack/inference/registry/ModelRegistry.java b/...gin/inference/src/main/java/org/elasticsearch/xpack/inference/registry/ModelRegistry.java
@@ -397,6 +397,7 @@ public void updateModelTransaction(Model newModel, Model existingModel, ActionLi
                 logger.error(
                     format("Failed to update inference endpoint [%s] due to [%s]", inferenceEntityId, configResponse.buildFailureMessage())
                 );
+                preventDeletionLock.remove(inferenceEntityId);
                 // Since none of our updates succeeded at this point, we can simply return.
                 finalListener.onFailure(
                     new ElasticsearchStatusException(