Replace loop in branch_graph.js with .map()

The following discussion from !20008 (merged) should be addressed:

• @kushalpandya started a discussion:

  While this MR is a good first iteration, I'd have preferred to see .map being used here in place of for loop.'

  @leiminghany Mind creating another MR as follow up while I merge this? 😄