Formatting

test/axon/compiler_test.exs

@@ -5371,52 +5371,6 @@ defmodule CompilerTest do
       assert map_size(params) == 1
     end
 
-    test "initializes correctly when using block from outside scope" do
-      block1 = Axon.block(&Axon.dense(&1, 32))
-
-      block2 =
-        Axon.block(fn x ->
-          x |> Axon.dense(32) |> block1.()
-        end)
-
-      model =
-        Axon.input("features")
-        |> Axon.dense(32)
-        |> block1.()
-        |> block2.()
-        |> block1.()
-        |> block2.()
-
-      {init_fn, _} = Axon.build(model)
-
-      assert %{
-               "block_0" => block_0_params,
-               "block_1" => block_1_params,
-               "dense_0" => dense_0_params
-             } = params = init_fn.(Nx.template({1, 1}, :f32), %{})
-
-      assert %{"dense_0" => %{"kernel" => b0k, "bias" => b0b}} = block_0_params
-      assert %{"dense_0" => %{"kernel" => b1k, "bias" => b1b}} = block_1_params
-      assert %{"kernel" => k1, "bias" => b1} = dense_0_params
-
-      assert Nx.shape(b0k) == {32, 32}
-      assert Nx.shape(b1k) == {32, 32}
-      assert Nx.shape(b0b) == {32}
-      assert Nx.shape(b1b) == {32}
-      assert Nx.shape(k1) == {1, 32}
-      assert Nx.shape(b1) == {32}
-      assert Nx.type(b0k) == {:f, 32}
-      assert Nx.type(b0b) == {:f, 32}
-      assert Nx.type(b1k) == {:f, 32}
-      assert Nx.type(b1b) == {:f, 32}
-      assert Nx.type(k1) == {:f, 32}
-      assert Nx.type(b1) == {:f, 32}
-
-      assert map_size(params) == 3
-      assert map_size(block_0_params) == 1
-      assert map_size(block_1_params) == 1
-    end
-
     test "predicts correctly with single dense, used once" do
       block = Axon.block(&Axon.dense(&1, 32))
       model = block.(Axon.input("features"))