<commit_before>set -exv

OPENMPI_DIR=/kubeflow/openmpi
BACKOFF_SECS=10
TIMEOUT_EXIT_CODE=124

wait_mpi_ready() {
  local workers=$1
  local max_retries=$2
  local retries=0

  until mpiexec -n ${workers} --hostfile ${OPENMPI_DIR}/assets/hostfile --allow-run-as-root -q sh -c 'echo $(hostname) is ready'; do
    sleep ${BACKOFF_SECS}<commit_msg>openmpi: add MPIEXEC_TIMEOUT for mpiexec in wait_mpi_ready() in init.shi (#773)<commit_after>set -exv

OPENMPI_DIR=/kubeflow/openmpi
MPIEXEC_TIMEOUT_ON_WAIT_MPI_READY=10
BACKOFF_SECS=10
TIMEOUT_EXIT_CODE=124

wait_mpi_ready() {
  local workers=$1
  local max_retries=$2
  local retries=0

  until MPIEXEC_TIMEOUT=${MPIEXEC_TIMEOUT_ON_WAIT_MPI_READY} mpiexec -n ${workers} --hostfile ${OPENMPI_DIR}/assets/hostfile --allow-run-as-root -q sh -c 'echo $(hostname) is ready'; do
    sleep ${BACKOFF_SECS}
{'fraction_non_alphanumeric': 0.11848341232227488, 'fraction_numerical': 0.024881516587677725, 'mean_word_length': 5.450381679389313, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '4645750', 'n_tokens_mistral': 353, 'n_tokens_neox': 337, 'n_words': 68}