Skip to content

Commit 96bfe11

Browse files
Nikhil BansalOrbax Authors
authored andcommitted
Pin xpk version in Orbax testing Dockerfile.
PiperOrigin-RevId: 892508771
1 parent 31f0377 commit 96bfe11

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

checkpoint/orbax/checkpoint/_src/testing/oss/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ RUN pip install --no-cache-dir gcsfs portpicker clu tensorflow pyyaml
4848

4949
# 4. Install Orbax from Source
5050
WORKDIR /app/orbax_repo/checkpoint
51-
RUN pip install xpk
51+
RUN pip install xpk==v1.0.0
5252

5353
# 5. Environment Setup
5454
# Set PYTHONPATH so 'import orbax' works from the correctly mapped directory

checkpoint/orbax/checkpoint/_src/testing/oss/cloud_run_integration_tests.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
tests:
22
- name: emergency_checkpoint_manager_benchmark
3-
cluster_name: orbax-cluster-test-mtc
3+
cluster_name: orbax-cluster-test-mtc-11
44
tpu_type: v5litepod-16
55
zone: us-west1-c
66
gcp_region: us-west1

0 commit comments

Comments
 (0)