if self._cluster_coordinator:
      self.test_function = lambda iterator: self._cluster_coordinator.schedule(  # pylint: disable=g-long-lambda
          test_function, args=(iterator,))