kokoro/pypi.cfg

# Copyright 2024 Google Inc. All Rights Reserved.
#
# 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.

build_file: "dataflux-pytorch/kokoro/pypi.sh"

env_vars {
  key: "PROJECT"
  value: "dataflux-project"
}

env_vars {
  key: "BUCKET"
  value: "official-dataflux-tess"
}

env_vars {
  key: "PREFIX"
  value: "UNet3D/micro/100KB-500MB/train"
}

env_vars {
  key: "LIST_WORKERS"
  value: "32"
}

env_vars {
  key: "FILE_COUNT"
  value: "5000"
}

env_vars {
  key: "TOTAL_FILE_SIZE"
  value: "501770000"
}

env_vars {
  key: "MAX_COMPOSE_BYTES"
  value: "100000000"
}

env_vars {
  key: "LIST_TIMEOUT"
  value: "10"
}

env_vars {
  key: "DOWNLOAD_TIMEOUT"
  value: "400"
}

env_vars {
  key: "PARALLELIZATION"
  value: "32"
}

action {
  define_artifacts {
    regex: "**/integration_tests/sponge_log.xml"
  }
}