Update continuous build script to tf_keras.

PiperOrigin-RevId: 566601750
keras-team · Sep 19, 2023 · 7f838e4 · 7f838e4
1 parent 773c5e6
commit 7f838e4
Show file tree

Hide file tree

Showing 6 changed files with 12 additions and 8 deletions.
diff --git a/tf_keras/kokoro/github/ubuntu/cpu/build.sh b/tf_keras/kokoro/github/ubuntu/cpu/build.sh
@@ -29,13 +29,15 @@ source venv/bin/activate
 python --version
 python3 --version
 
-cd "src/github/keras"
+cd "src/github/tf-keras"
 
 # Keep pip version at 20.1.1 to avoid the slow resolver issue.
 pip install -U pip==20.1.1 setuptools
 pip install -r requirements.txt
 # Uninstall the keras-nightly package so that we will only test the version of
 # keras code from local workspace.
+# TODO(keras-team): `tf-nightly` currently installs `keras-nightly`.
+# Update this once we switch to `tf_keras-nightly` in TensorFlow.
 pip uninstall -y keras-nightly
 
 # TODO(scottzhu): Using --define=use_fast_cpp_protos=false to suppress the
@@ -45,4 +47,4 @@ bazel test --test_timeout 300,450,1200,3600 --test_output=errors --keep_going \
    --build_tests_only \
    --build_tag_filters="-no_oss,-oss_excluded" \
    --test_tag_filters="-no_oss,-oss_excluded" \
-   -- //keras/...
+   -- //tf_keras/...
diff --git a/tf_keras/kokoro/github/ubuntu/cpu/continuous.cfg b/tf_keras/kokoro/github/ubuntu/cpu/continuous.cfg
@@ -1,4 +1,4 @@
-build_file: "keras/keras/kokoro/github/ubuntu/cpu/build.sh"
+build_file: "tf-keras/tf_keras/kokoro/github/ubuntu/cpu/build.sh"
 
 action {
   define_artifacts {

diff --git a/tf_keras/kokoro/github/ubuntu/cpu/presubmit.cfg b/tf_keras/kokoro/github/ubuntu/cpu/presubmit.cfg
@@ -1,4 +1,4 @@
-build_file: "keras/keras/kokoro/github/ubuntu/cpu/build.sh"
+build_file: "tf-keras/tf_keras/kokoro/github/ubuntu/cpu/build.sh"
 
 action {
   define_artifacts {

diff --git a/tf_keras/kokoro/github/ubuntu/gpu/build.sh b/tf_keras/kokoro/github/ubuntu/gpu/build.sh
@@ -29,13 +29,15 @@ source venv/bin/activate
 python --version
 python3 --version
 
-cd "src/github/keras"
+cd "src/github/tf-keras"
 
 # Keep pip version at 20.1.1 to avoid the slow resolver issue.
 pip install -U pip==20.1.1 setuptools
 pip install -r requirements.txt
 # Uninstall the keras-nightly package so that we will only test the version of
 # keras code from local workspace.
+# TODO(keras-team): `tf-nightly` currently installs `keras-nightly`.
+# Update this once we switch to `tf_keras-nightly` in TensorFlow.
 pip uninstall -y keras-nightly
 
 # LD Library Path needs to be same as TensorFlow Ubuntu Docker build -
@@ -67,4 +69,4 @@ bazel test --test_timeout 300,600,1200,3600 --test_output=errors --keep_going \
    --test_tag_filters="${tag_filters}" \
    --run_under=@org_keras//keras/tools/gpu_build:parallel_gpu_execute \
    --local_test_jobs=${LOCAL_TEST_JOBS} \
-   -- //keras/...
+   -- //tf_keras/...
diff --git a/tf_keras/kokoro/github/ubuntu/gpu/continuous.cfg b/tf_keras/kokoro/github/ubuntu/gpu/continuous.cfg
@@ -1,4 +1,4 @@
-build_file: "keras/keras/kokoro/github/ubuntu/gpu/build.sh"
+build_file: "tf-keras/tf_keras/kokoro/github/ubuntu/gpu/build.sh"
 
 action {
   define_artifacts {

diff --git a/tf_keras/kokoro/github/ubuntu/gpu/presubmit.cfg b/tf_keras/kokoro/github/ubuntu/gpu/presubmit.cfg
@@ -1,4 +1,4 @@
-build_file: "keras/keras/kokoro/github/ubuntu/gpu/build.sh"
+build_file: "tf-keras/tf_keras/kokoro/github/ubuntu/gpu/build.sh"
 
 action {
   define_artifacts {