Fix deprecated use of Tuple.append/2

Author: jonatanklosko

lib/bumblebee/diffusion/layers/unet.ex

@@ -123,7 +123,8 @@ defmodule Bumblebee.Diffusion.Layers.UNet do
               hidden_state
             end
 
-          {hidden_state, Tuple.append(output_states, {hidden_state, out_channels})}
+          {hidden_state,
+           Tuple.insert_at(output_states, tuple_size(output_states), {hidden_state, out_channels})}
       end
 
     if add_downsample do
@@ -133,7 +134,8 @@ defmodule Bumblebee.Diffusion.Layers.UNet do
           name: join(name, "downsamples.0")
         )
 
-      {hidden_state, Tuple.append(output_states, {hidden_state, out_channels})}
+      {hidden_state,
+       Tuple.insert_at(output_states, tuple_size(output_states), {hidden_state, out_channels})}
     else
       {hidden_state, output_states}
     end

lib/bumblebee/layers.ex

@@ -923,7 +923,13 @@ defmodule Bumblebee.Layers do
   Appends tuple element to the node result.
   """
   def append(%Axon{} = tuple, %Axon{} = x) do
-    Axon.layer(fn tuple, x, _ -> Tuple.append(tuple, x) end, [tuple, x], op_name: :append)
+    Axon.layer(
+      fn tuple, x, _ ->
+        Tuple.insert_at(tuple, tuple_size(tuple), x)
+      end,
+      [tuple, x],
+      op_name: :append
+    )
   end
 
   @doc """

lib/bumblebee/multimodal/layout_lm.ex

@@ -272,7 +272,7 @@ defmodule Bumblebee.Multimodal.LayoutLm do
     shape = Keyword.get(opts, :shape, {nil, nil})
 
     attention_head_mask_shape = {spec.num_blocks, spec.num_attention_heads}
-    bounding_box_shape = Tuple.append(shape, 4)
+    bounding_box_shape = Tuple.insert_at(shape, tuple_size(shape), 4)
 
     Bumblebee.Utils.Model.inputs_to_map([
       Axon.input("input_ids", shape: shape),

lib/bumblebee/text/bert.ex

@@ -398,7 +398,7 @@ defmodule Bumblebee.Text.Bert do
     shape = Keyword.get(opts, :shape, {nil, nil})
     decoder? = Keyword.get(opts, :decoder?, false)
 
-    hidden_shape = Tuple.append(shape, spec.hidden_size)
+    hidden_shape = Tuple.insert_at(shape, tuple_size(shape), spec.hidden_size)
     attention_head_mask_shape = {spec.num_blocks, spec.num_attention_heads}
 
     inputs =

lib/bumblebee/text/roberta.ex

@@ -354,7 +354,7 @@ defmodule Bumblebee.Text.Roberta do
     shape = Keyword.get(opts, :shape, {nil, nil})
     decoder? = Keyword.get(opts, :decoder?, false)
 
-    hidden_shape = Tuple.append(shape, spec.hidden_size)
+    hidden_shape = Tuple.insert_at(shape, tuple_size(shape), spec.hidden_size)
     attention_head_mask_shape = {spec.num_blocks, spec.num_attention_heads}
 
     inputs =