From 5081978346835acf911a224a8aa647802656239e Mon Sep 17 00:00:00 2001
From: Dmitriy Alekseev <1865999+dragoangel@users.noreply.github.com>
Date: Thu, 22 Jan 2026 22:14:42 +0100
Subject: [PATCH] Fix after merging master

---
 src/plugins/lua/neural.lua | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/plugins/lua/neural.lua b/src/plugins/lua/neural.lua
index ad0ef94ce6..17705d4cf4 100644
--- a/src/plugins/lua/neural.lua
+++ b/src/plugins/lua/neural.lua
@@ -850,6 +850,13 @@ local function maybe_train_existing_ann(worker, ev_base, rule, set, profiles)
   if sel_elt then
     -- We have our ANN and that's train vectors, check if we can learn
     local ann_key = sel_elt.redis_key
+
+    -- Check if we need to train ann
+    if rule.train.store_set_only then
+      lua_util.debugm(N, rspamd_config, "skiped check if ANN %s needs to be trained due to store_set_only", ann_key)
+      return
+    end
+
     local pending_key = neural_common.pending_train_key(rule, set)
 
     lua_util.debugm(N, rspamd_config, "check if ANN %s (pending %s) needs to be trained",
-- 
2.47.3