This should work, but is longer than Flow.map version from book:
...
|> Flow.reject(&(&1.type == "closed"))
|> Flow.partition(key: {:key, :country})
|> Flow.group_by(& &1.country)
|> Flow.on_trigger(fn map ->
country_data = Enum.map(map, fn {country, data} -> {country, Enum.count(data)} end)
{country_data, map}
end)
#|> Flow.map(fn {country, data} -> {country, Enum.count(data)} end)
|> Enum.to_list()