Machine Learning in Elixir: Chapter 6 Axon.Display.as_table does not work (Page 133)

Updating to latest versions does not correct the error.

Mix.install([
  {:axon, "~> 0.7.0"},
  {:nx, "~> 0.9.2"},
  {:exla, "~> 0.9.2"},
  {:stb_image, "~> 0.6.10"},
  {:kino, "~> 0.17.0"},
  {:table_rex, "~> 3.1.1"}
])

Anyone have any ideas what else to try?