.travis.yml

language: python
python: 3.9

env:
  global:
  - LOCAL_REPO: transformers-inference
  - REMOTE_REPO: semitechnologies/transformers-inference
  - secure: QLL5aAA41AjtgVeHPmgmtpp/nOup0R5FT/wIJwPEAjIWl2MuOCnqaHq5pE+XM80tkbAHhVz48IyLH6P5FkcYQO0dmRHO9M6n3jawrrsTvbQ1dIMqEarl69mO7+xQZJS/ggN6E8Q0Q5Y0Q8f/btC6osHr01OOyqd+Tb3Bcihv/zFmPydULSAu2pL4n89PkY52PjHgxkrsBBxiN3O95q/FwGCVWq+kH1bSkEAAl/miXA0wCss4I108cUmoY4RS1CYKAGzRk0fK6d+r+sIw08pX3Gvr/2KFQhCK0mJAPKnBZ/VBDFW1ilZ1LZseciYSX/RtHYvWgTb1xemAR3ewmXmJYP9mqvaBArleZARg4cWsz7Qz9Mj9oiC4EGgCxHrzXBV2d+SwIh1JFs3k87P+DUyDTmoR6rTTMMQ4lLPnPSRj0lQtvCAl861ojivkQG4jOKKYyJiKKDmTqGsJ6Ys53MNkzsHCUEZCZxJlSMqOLZGxr+U/A7F8LCUt/h2rgyL9QT1kCz2tNYyE3BMktAyKeLD4np7cHxkLUbPv59HkEffCfP/79B+1s2Gb3Bok9F/u1oiIXJ3pO3gaFXfN4/RxG7lROpNMhRin4yo7YU/kBNu432b4ufcJ1LPmqWAH+FN/0OidZPxitrHWPPklHga718/yhl+mkObWm9uJmt0zmK+ZzIM=
  - secure: Kk1k4is7vziUkv6ZJqbIuY6zQCesWYILmI4x9MHQcwSzmrKas5fSvSFobQZHKp9IdpUQ6awR18+yJHmqU+I25LqCm3AjvSz2OmS3d/Fq/Q3U5C7mRrSprQT/zjCiTwjjrc7xO9RaaLKX92dHZkHvmYk2jf/1F3gMZGw1RpdTfj+pUbgIvw/AL8CaIPkNKZpR0tjpm3fOIO4MCgu0JpzscnCF00ANKc/Gy2jtmts8ee1lmnLRBeci1cRwbJOqzHh8AhJG3rrVCu9ISiqd62vOXlM+z7W2KUdTtVo5bNBa5iBuBa+kwd/uH95p9NVdMuISaO+cj5fT9sFMLPpLcAjnLjChGMBUIeT5pSedWwziYj48TZqIEgmza4iBwVyXIBx6Jblml+j8EobnDtQCtHbHU98FgBO5AdaNbt4laHlOQCSs2Kt2AFT6UAS8NqT842gHoG9jr6ovYpwb27TjE8ILCnCRnhy9e3VW0VCRHCrgotwCAV/gM4S0kVaO5IAa8vxDp0w+4ewdh60aysg4p6us21Iim746WNi4xXPzkeJ/ce/X1abDy2yMc6XP3el4OGIYe701/wDblcmjybNqq+RjphtwmNJUnk+BdykW+006+xkzAo58IcFZMqkeDFd/FOa5OEM6w1HA1rXM5IYGWwrQudx7lyFOzPAxPt938Gk33UE=

install: echo skipping default travis install script
  
jobs:
  include:
    # always run test, etc. for this model
    - env: 
        MODEL_NAME: distilbert-base-uncased
      stage: buildanddeploy

    # to prevent spamming the pipeline, only include most models if there is a tag
    #
    # TODO: should this be deprecated or not?
    - env: 
        MODEL_NAME: sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2
        MODEL_TAG_NAME: sentence-transformers-paraphrase-multilingual-MiniLM-L12-v2
      if: tag IS present
      stage: buildanddeploy
    - env: 
        MODEL_NAME: sentence-transformers/multi-qa-MiniLM-L6-cos-v1
        MODEL_TAG_NAME: sentence-transformers-multi-qa-MiniLM-L6-cos-v1
      if: tag IS present
      stage: buildanddeploy
    - env: 
        MODEL_NAME: sentence-transformers/multi-qa-mpnet-base-cos-v1
        MODEL_TAG_NAME: sentence-transformers-multi-qa-mpnet-base-cos-v1
      if: tag IS present
      stage: buildanddeploy
    - env: 
        MODEL_NAME: sentence-transformers/all-mpnet-base-v2 
        MODEL_TAG_NAME: sentence-transformers-all-mpnet-base-v2 
      if: tag IS present
      stage: buildanddeploy
    - env: 
        MODEL_NAME: sentence-transformers/all-MiniLM-L12-v2  
        MODEL_TAG_NAME: sentence-transformers-all-MiniLM-L12-v2   
      if: tag IS present
      stage: buildanddeploy
    - env: 
        MODEL_NAME: sentence-transformers/paraphrase-multilingual-mpnet-base-v2
        MODEL_TAG_NAME: sentence-transformers-paraphrase-multilingual-mpnet-base-v2
      if: tag IS present
      stage: buildanddeploy
    - env:
        MODEL_NAME: sentence-transformers/gtr-t5-base
        MODEL_TAG_NAME: sentence-transformers-gtr-t5-base
      if: tag IS present
      stage: buildanddeploy
    - env:
        MODEL_NAME: sentence-transformers/gtr-t5-large
        MODEL_TAG_NAME: sentence-transformers-gtr-t5-large
      if: tag IS present
      stage: buildanddeploy
    - env:
        MODEL_NAME: vblagoje/dpr-ctx_encoder-single-lfqa-wiki
        MODEL_TAG_NAME: vblagoje-dpr-ctx_encoder-single-lfqa-wiki
      if: tag IS present
      stage: buildanddeploy
    - env:
        MODEL_NAME: vblagoje/dpr-question_encoder-single-lfqa-wiki
        MODEL_TAG_NAME: vblagoje-dpr-question_encoder-single-lfqa-wiki
      if: tag IS present
      stage: buildanddeploy
    - env:
        MODEL_NAME: facebook/dpr-ctx_encoder-single-nq-base
        MODEL_TAG_NAME: facebook-dpr-ctx_encoder-single-nq-base
      if: tag IS present
      stage: buildanddeploy
    - env:
        MODEL_NAME: facebook/dpr-question_encoder-single-nq-base
        MODEL_TAG_NAME: facebook-dpr-question_encoder-single-nq-base
      if: tag IS present
      stage: buildanddeploy
    # Disabled due to memory exaustion in Travis, configuration options can be found here:
    # https://docs.travis-ci.com/user/enterprise/worker-configuration/
    # - env:
    #     MODEL_NAME: sentence-transformers/gtr-t5-xl
    #     MODEL_TAG_NAME: sentence-transformers-gtr-t5-xl
    #   if: tag IS present
    #   stage: buildanddeploy

    # no longer build the old models as the pipeline is already slow enough.
    # They will still be accessible through Dockerhub, but will no longer be
    # updated.
    # - env: 
    #     MODEL_NAME: sentence-transformers/paraphrase-mpnet-base-v2
    #     MODEL_TAG_NAME: sentence-transformers-paraphrase-mpnet-base-v2
    #   if: tag IS present
    #   stage: buildanddeploy
    # - env: 
    #     MODEL_NAME: sentence-transformers/paraphrase-distilroberta-base-v2
    #     MODEL_TAG_NAME: sentence-transformers-paraphrase-distilroberta-base-v2
    #   if: tag IS present
    #   stage: buildanddeploy
    # - env: 
    #     MODEL_NAME: sentence-transformers/stsb-mpnet-base-v2
    #     MODEL_TAG_NAME: sentence-transformers-stsb-mpnet-base-v2
    #   if: tag IS present
    #   stage: buildanddeploy
    # - env: 
    #     MODEL_NAME: sentence-transformers/nli-mpnet-base-v2
    #     MODEL_TAG_NAME: sentence-transformers-nli-mpnet-base-v2
    #   if: tag IS present
    #   stage: buildanddeploy
    # - env: 
    #     MODEL_NAME: sentence-transformers/msmarco-distilbert-base-v3
    #     MODEL_TAG_NAME: sentence-transformers-msmarco-distilbert-base-v3
    #   if: tag IS present
    #   stage: buildanddeploy
    # - env: 
    #     MODEL_NAME: sentence-transformers/stsb-distilbert-base 
    #     MODEL_TAG_NAME: sentence-transformers-stsb-distilbert-base
    #   if: tag IS present
    #   stage: buildanddeploy
    # - env: 
    #     MODEL_NAME: sentence-transformers/quora-distilbert-base 
    #     MODEL_TAG_NAME: sentence-transformers-quora-distilbert-base
    #   if: tag IS present
    #   stage: buildanddeploy
    # - env: 
    #     MODEL_NAME: sentence-transformers/paraphrase-distilroberta-base-v1 
    #     MODEL_TAG_NAME: sentence-transformers-paraphrase-distilroberta-base-v1
    #   if: tag IS present
    #   stage: buildanddeploy
    # - env: 
    #     MODEL_NAME: kiri-ai/distiluse-base-multilingual-cased-et 
    #     MODEL_TAG_NAME: kiri-ai-distiluse-base-multilingual-cased-et
    #   if: tag IS present
    #   stage: buildanddeploy
    # - env: 
    #     MODEL_NAME: sentence-transformers/msmarco-distilroberta-base-v2 
    #     MODEL_TAG_NAME: sentence-transformers-msmarco-distilroberta-base-v2
    #   if: tag IS present
    #   stage: buildanddeploy
    # - env: 
    #     MODEL_NAME: sentence-transformers/msmarco-distilbert-base-v2 
    #     MODEL_TAG_NAME: sentence-transformers-msmarco-distilbert-base-v2
    #   if: tag IS present
    #   stage: buildanddeploy
    # - env: 
    #     MODEL_NAME: sentence-transformers/stsb-xlm-r-multilingual 
    #     MODEL_TAG_NAME: sentence-transformers-stsb-xlm-r-multilingual
    #   if: tag IS present
    #   stage: buildanddeploy
    # - env: 
    #     MODEL_NAME: sentence-transformers/paraphrase-xlm-r-multilingual-v1 
    #     MODEL_TAG_NAME: sentence-transformers-paraphrase-xlm-r-multilingual-v1
    #   if: tag IS present
    #   stage: buildanddeploy

    - name: Build the base for custom images
      if: env(DOCKER_PASSWORD) IS present
      script: GIT_BRANCH=$TRAVIS_BRANCH GIT_TAG=$TRAVIS_TAG GIT_PULL_REQUEST=$TRAVIS_PULL_REQUEST cicd/build_custom_base.sh
      stage: postbuild

script:
- cicd/build.sh || travis_terminate 1
- cicd/test.sh || travis_terminate 1
- |-
  if [ -n "$DOCKER_PASSWORD" ]
  then
    # Push docker image
    GIT_BRANCH=$TRAVIS_BRANCH \
    GIT_TAG=$TRAVIS_TAG \
    GIT_PULL_REQUEST=$TRAVIS_PULL_REQUEST \
    travis_wait 120 cicd/docker_push.sh || travis_terminate 1
  else
    # Skip pushing docker image
    echo "Skipping docker image push step. Tests passed."
    travis_terminate 0
  fi

before_install: 
  # install a newer docker version which supports buildx
  - sudo rm -rf /var/lib/apt/lists/*
  - curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
  - sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) edge"
  - sudo apt-get update
  - sudo apt-get -y -o Dpkg::Options::="--force-confnew" install docker-ce
  - mkdir -vp ~/.docker/cli-plugins/
  - curl --silent -L "https://github.com/docker/buildx/releases/download/v0.3.0/buildx-v0.3.0.linux-amd64" > ~/.docker/cli-plugins/docker-buildx
  - chmod a+x ~/.docker/cli-plugins/docker-buildx

before_script:
  # if there is a password provided login to docker at the very beginning, so we don't run into rate-limiting
- |-
  if [ -n "$DOCKER_PASSWORD" ]
  then
    echo "$DOCKER_PASSWORD" | docker login -u "$DOCKER_USERNAME" --password-stdin
  fi

# currently the latest available - so we can install the latest docker
dist: bionic