Use a Typespec function type

You can write this spec, it just doesn’t mean what you want:

defmodule B do
  @spec my_fun :: A.simple_function()
  def my_fun do
    fn
      x when x >= 0 -> x + 1
      x -> -x + 1
    end
  end
end