message: Adjust after streams change
  Summary: Ref T16509
  Test Plan: All tests pass
  Reviewers: jakeh, hakons, #tensorflow, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved
  Maniphest Tasks: T16509
diff: @@ -106,7 +106,7 @@ class CombinedWeightsTest(xla_test.XLATestCase):
  'Switch')['children'][0]
  self.assertLess(
  - len(report.get_program(download_weights_index)['children']), 8,
  + len(report.get_program(download_weights_index)['children']), 11,
  "The download weights sequence should not have lots of entries "
  "(because the copies will have been merged)")
  