diff --git a/test/tpu/xla_test_job.yaml b/test/tpu/xla_test_job.yaml index 06fe993b51c8..faf2d67ba08c 100644 --- a/test/tpu/xla_test_job.yaml +++ b/test/tpu/xla_test_job.yaml @@ -41,7 +41,7 @@ spec: - bash - -cxe - | - pip install expecttest + pip install expecttest==0.1.6 pip install rich python3 /src/pytorch/xla/test/test_operations.py -v