diff --git a/test/tpu/xla_test_job.yaml b/test/tpu/xla_test_job.yaml index e727953ddc4..2e74cc668d0 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