i'm jet-lagged

NVIDIA · Nov 13, 2023 · 12a57eb · 12a57eb
1 parent 602002f
commit 12a57eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/container/Dockerfile.pax.amd64 b/.github/container/Dockerfile.pax.amd64
@@ -21,7 +21,7 @@ ARG SRC_PATH_PAXML
 ARG SRC_PATH_PRAXIS
 
 # update TE manifest file to install the [test] extras
-RUN sed -i "s/.whl/.whl[test]/g" /opt/pip-tools.d/manifest.te
+RUN sed -i "s/transformer-engine @/transformer-engine[test] @/g" /opt/pip-tools.d/manifest.te
 
 RUN <<"EOF" bash -ex
 get-source.sh -f ${REPO_PAXML}  -r ${REF_PAXML}  -d ${SRC_PATH_PAXML}