docker.sh

#!/bin/sh

# Example search: ./docker.sh fast search seed_datasets_current/185_baseball/ alphad3m:latest
# Example ta2ta3: ./docker.sh ta2ta3 seed_datasets_current/185_baseball/ alphad3m:latest

# You should have defined these enviroment variables: $LOCAL_INPUT_ROOT, $LOCAL_OUTPUT_ROOT and $LOCAL_STATIC_ROOT

set -eu

OPTS=""
TIMEOUT=10
PORT=45042

while true; do
    if [ "$1" = "fast" ]; then
        OPTS="$OPTS -e TA2_DEBUG_BE_FAST=1"
        TIMEOUT=1
        shift
    else
        break
    fi
done
case "$1" in
    search)
        MODE=search
        INPUT="$2"
        shift 2
    ;;
    ta2ta3)
        MODE=ta2ta3
        INPUT="$2"
        shift 2
    ;;
    *)
        echo "Usage:\n  $(basename $0) search seed_datasets_current/185_baseball/TRAIN <image>" >&2
        echo "  $(basename $0) ta2ta3 seed_datasets_current/185_baseball/TRAIN <image>" >&2
        echo "  $(basename $0) test seed_datasets_current/185_baseball/TEST 0123-4567-89abcdef <image>" >&2
        exit 2
    ;;
esac

docker run -ti --rm \
    -p ${PORT}:45042 \
    -e D3MRUN="$MODE" \
    -e D3MINPUTDIR=/input \
    -e D3MOUTPUTDIR=/output \
    -e D3MSTATICDIR=/static \
    -e D3MCPU=4 \
    -e D3MRAM=4Gi \
    -e D3MTIMEOUT=$TIMEOUT \
    -e D3MPORT=$PORT \
    $OPTS \
    -v "$PWD/alphad3m/alphad3m:/usr/src/app/alphad3m" \
    -v "$PWD/tests:/usr/src/app/tests"\
    -v "$PWD/eval.sh:/usr/local/bin/eval.sh"\
    -v "$LOCAL_INPUT_ROOT/${INPUT}:/input" \
    -v "$LOCAL_OUTPUT_ROOT:/output" \
    -v "$LOCAL_STATIC_ROOT:/static" \
    --name ta2_container \
    "$@"