#!/bin/bash

HOME_DIR=/test
BIN_DIR=${HOME_DIR}/bin
LOG_DIR=${HOME_DIR}/logs

if [[ $(python -c 'import tensorflow as tf; print(tf.__version__)') == 2.0.* ]]; then
  pip install -U --no-deps tensorflow-addons==0.6.0
elif [[ $(python -c 'import tensorflow as tf; print(tf.__version__)') == 2.1.* && $(python -c 'import sys; print(".".join(map(str, sys.version_info[:1])))') == 3 ]]; then
  pip install -U --no-deps tensorflow-addons==0.9.0
elif [[ $(python -c 'import tensorflow as tf; print(tf.__version__)') == 2.2.* ]]; then
  pip install -U --no-deps tensorflow-addons==0.10.0
else
  pip install -U --no-deps tensorflow_addons
fi
pip install typeguard packaging

python ${BIN_DIR}/testTFAddons.py || exit 1

exit 0