WAR the bug in t5x dependency. It currently need the dev version of j…

…estimator.
jax-ml · May 30, 2023 · cdced24 · cdced24
1 parent acfeb9b
commit cdced24
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/slurm_job_scripts/run_e2e_t5x_tests.sub b/.github/workflows/slurm_job_scripts/run_e2e_t5x_tests.sub
@@ -53,6 +53,7 @@ rm -rf ${E2E_TESTS_WORKSPACE_DIR}/* \
 && python3.8 -m pip uninstall -y cudf dask-cudf \
 && python3.8 -m pip install ${T5X_DIR} \
 && python3.8 -m pip install ${T5X_DIR}/t5x/contrib/gpu/scripts_gpu/dummy_wikipedia_config \
+&& python3.8 -m pip install -U git+https://github.com/google-research/jestimator \
 && hostname > ${E2E_TESTS_WORKSPACE_DIR}/hostname.txt
 EOF