original_docstring: 
          Return a list of all Units in a terminal completed state from all
          the provided TaskRuns.
          
docstring: Return a list of all Units in a terminal completed state from all
  the provided TaskRuns.
docstring_tokens: Return a list of all Units in a terminal completed state from all the provided TaskRuns .
code: def _get_units_for_task_runs(self, task_runs: List[TaskRun]) -> List[Unit]:
          """
          Return a list of all Units in a terminal completed state from all
          the provided TaskRuns.
          """
          units = []
          for task_run in task_runs:
              assignments = task_run.get_assignments()
              for assignment in assignments:
                  found_units = assignment.get_units()
                  for unit in found_units:
                      if unit.get_status() in [
                          AssignmentState.COMPLETED,
                          AssignmentState.ACCEPTED,
                          AssignmentState.REJECTED,
                          AssignmentState.SOFT_REJECTED,
                      ]:
                          units.append(unit)
          return units
code_tokens: def _get_units_for_task_runs ( self , task_runs : List [ TaskRun ] ) -> List [ Unit ] : units = [ ] for task_run in task_runs : assignments = task_run . get_assignments ( ) for assignment in assignments : found_units = assignment . get_units ( ) for unit in found_units : if unit . get_status ( ) in [ AssignmentState . COMPLETED , AssignmentState . ACCEPTED , AssignmentState . REJECTED , AssignmentState . SOFT_REJECTED , ] : units . append ( unit ) return units
short_docstring: Return a list of all Units in a terminal completed state from all
  the provided TaskRuns.
short_docstring_tokens: Return a list of all Units in a terminal completed state from all the provided TaskRuns .
comment: """
          Return a list of all Units in a terminal completed state from all
          the provided TaskRuns.
          """