Handle empty list of entities when aggregating entities (#386)

elixir-nx · Aug 5, 2024 · 5245a7e · 5245a7e
1 parent 78a7dcf
commit 5245a7e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/bumblebee/text/token_classification.ex b/lib/bumblebee/text/token_classification.ex
@@ -197,6 +197,8 @@ defmodule Bumblebee.Text.TokenClassification do
     end)
   end
 
+  defp override_token_labels([], _spec, _strategy), do: []
+
   defp override_token_labels([entity | entities], spec, strategy) do
     {final_group, word_entities} =
       Enum.reduce(entities, {[entity], []}, fn entity, {word_group, word_entities} ->