fix: Improve logic around grid.get_branches() for better performance

fix: Improve logic around grid.get_branches() for better performance #960