docker-compose.yml

# Copyright 2019 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

version: '3.1'

# Extensions are not support until 3.4 thus the repeated boilerplate below.

# We mount the struct2tensor project root at /struct2tensor (which is the WORKDIR of the image)
# in the container.
#
# We need to set PYTHON_VERSION in `build` command which is used to pick the
# base TF SIG Build image.
# The TF_VERSION environment variables is expected when run the pipeline
# Valid TF_VERSION are NIGHTLY_TF|NIGHTLY_TF_2|RELEASED_TF|PRERELEASED_TF|RELEASED_TF_2|PRERELEASED_TF_2
# Sample usage:
# docker-compose build --build-arg PYTHON_VERSION=3.7 manylinux2014
# docker-compose run -e TF_VERSION=NIGHTLY_TF manylinux2014

services:
  manylinux2014:
    image: struct2tensor-build:manylinux2014
    build:
      context: .
      dockerfile: struct2tensor/tools/docker_build/Dockerfile.manylinux2014
    volumes:
      - .:/struct2tensor:delegated