#!/usr/bin/env sh

set -o errexit
set -o nounset

LOG_PIPE="$(mktemp -u)"
mkfifo "$LOG_PIPE"

(
  while IFS= read -r line; do
    if echo "$line" | grep -qEi "\[(WARN|ERROR)\]"; then
      echo "$line" >&2
    else
      echo "$line"
    fi
  done < "$LOG_PIPE"
) &

exec machine-learning "$@" > "$LOG_PIPE" 2>&1