def _create_taskgroup()

in python-batch/batch.py [0:0]


  def _create_taskgroup(self):
      """ 
        Tasks are grouped inside a job using TaskGroups.
        Currently, it's possible to have only one task group.
      """

      self.group = batch_v1.TaskGroup()
      self.group.task_spec = self.task
      self.group.task_count = self.config["task_count"] if "task_count" in self.config else 1
      self.group.parallelism = self.config["parallelism"] if "parallelism" in self.config else 1
      self.group.task_count_per_node = self.config["task_count_per_node"] if "task_count_per_node" in self.config else 1

      return(self.group)