From 3140298dfcc059a24640bf4e99ee7686bab338d2 Mon Sep 17 00:00:00 2001
From: Max Hniebergall <137079448+maxhniebergall@users.noreply.github.com>
Date: Thu, 17 Oct 2024 10:25:37 -0400
Subject: [PATCH] Add missing preventDeletionLock.remove in corner case
 (#115010)

(cherry picked from commit 831d55fcc8354a1798c243f90bc5b0998158d2ae)
---
 .../elasticsearch/xpack/inference/registry/ModelRegistry.java    | 1 +
 1 file changed, 1 insertion(+)

diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/registry/ModelRegistry.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/registry/ModelRegistry.java
index 33a97f1e91621..260d4e663dafd 100644
--- a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/registry/ModelRegistry.java
+++ b/x-pack/plugin/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(