Create and use 2.17 toolchains in Docker build containers.

PiperOrigin-RevId: 623491147
tensorflow · Apr 10, 2024 · 1961f90 · 1961f90
1 parent 8520e31
commit 1961f90
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/tensorflow_decision_forests/keras/BUILD b/tensorflow_decision_forests/keras/BUILD
@@ -1,5 +1,5 @@
-load("@rules_python//python:defs.bzl", "py_binary", "py_library")
 load("@org_tensorflow//tensorflow:tensorflow.bzl", "tf_cc_binary")
+load("@rules_python//python:defs.bzl", "py_binary", "py_library")
 load("//tensorflow_decision_forests/keras/wrapper:wrapper.bzl", "py_wrap_yggdrasil_learners")
 
 package(

diff --git a/tensorflow_decision_forests/tensorflow/ops/inference/BUILD b/tensorflow_decision_forests/tensorflow/ops/inference/BUILD
@@ -1,8 +1,8 @@
-load("@rules_python//python:defs.bzl", "py_library")
 load(
     "@org_tensorflow//tensorflow:tensorflow.bzl",
     "tf_gen_op_wrapper_py",
 )
+load("@rules_python//python:defs.bzl", "py_library")
 
 package(
     default_visibility = ["//visibility:public"],

diff --git a/tensorflow_decision_forests/tensorflow/ops/training/BUILD b/tensorflow_decision_forests/tensorflow/ops/training/BUILD
@@ -1,5 +1,5 @@
-load("@rules_python//python:defs.bzl", "py_library")
 load("@org_tensorflow//tensorflow:tensorflow.bzl", "tf_gen_op_wrapper_py")
+load("@rules_python//python:defs.bzl", "py_library")
 
 package(
     default_visibility = ["//visibility:public"],

diff --git a/tools/start_compile_docker.sh b/tools/start_compile_docker.sh
@@ -58,7 +58,7 @@
 #  ./tools/build_pip_package.sh ALL_VERSIONS_ALREADY_ASSEMBLED
 #
 # https://hub.docker.com/r/tensorflow/build/tags?page=1
-DOCKER=tensorflow/build:2.16-python3.9
+DOCKER=tensorflow/build:2.17-python3.9
 
 # Current directory
 # Useful if Yggdrasil Decision Forests is available locally in a neighbor