diff --git a/tests/tensorflow/r2.12/common.libsonnet b/tests/tensorflow/r2.12/common.libsonnet
index 1aa253f86..ae0ae75d8 100644
--- a/tests/tensorflow/r2.12/common.libsonnet
+++ b/tests/tensorflow/r2.12/common.libsonnet
@@ -65,6 +65,7 @@ local mixins = import 'templates/mixins.libsonnet';
                   import tensorflow as tf
                   import urllib
                   import json
+                  os.system('pip3 install cloud_tpu_client')
                   import cloud_tpu_client
                   import sys
                   print('python version: ' + str(sys.version))