Enable distribution test (#19947)

* Enable distribution test. * Run distribution test separately.
2024-07-03 14:06:23 -04:00 · 2024-07-03 14:06:23 -04:00 · 13c72f9180
commit 13c72f9180
parent d910dcb827
1 changed files with 2 additions and 0 deletions
--- a/.kokoro/github/ubuntu/gpu/build.sh
+++ b/.kokoro/github/ubuntu/gpu/build.sh
@ -57,6 +57,8 @@ then
               --ignore keras/src/backend/jax/distribution_lib_test.py \
               --ignore keras/src/distribution/distribution_lib_test.py \
               --cov=keras
+
+   pytest keras/src/distribution/distribution_lib_test.py --cov=keras
 fi

 if [ "$KERAS_BACKEND" == "torch" ]