FIX: Ensure that we shutdown thread pool (#1207)

2025-03-21 11:08:36 +08:00 · 2025-03-21 11:08:36 +08:00 · 01eced74a3
parent 451f7662b2
commit 01eced74a3
2 changed files with 2 additions and 2 deletions
--- a/lib/embeddings/vector.rb
+++ b/lib/embeddings/vector.rb
@ -49,7 +49,7 @@ module DiscourseAi
          .zip(*promised_embeddings)
          .value!
          .each { |e| schema.store(e[:target], e[:embedding], e[:digest]) }
-
+      ensure
        pool.shutdown
        pool.wait_for_termination
      end
--- a/lib/sentiment/post_classification.rb
+++ b/lib/sentiment/post_classification.rb
@ -92,7 +92,7 @@ module DiscourseAi
          .zip(*promised_classifications)
          .value!
          .each { |r| store_classification(r[:target], r[:classification]) }
-
+      ensure
        pool.shutdown
        pool.wait_for_termination
      end