From 074e22ff5d442e725eb7b35c05758274203e9527 Mon Sep 17 00:00:00 2001
From: Max Hniebergall <max.hniebergall@elastic.co>
Date: Thu, 17 Oct 2024 09:15:54 -0400
Subject: [PATCH] Add missing preventDeletionLock.remove in corner case

---
 .../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(