Replace zip/unzip command with wheel pack/unpack.
PiperOrigin-RevId: 329453192 Change-Id: I4967c8a5862399cfab9a98fe68dc2e40df3c35a9
This commit is contained in:
parent
dac18d7a05
commit
a272ba2ec4
@ -260,22 +260,22 @@ function copy_to_new_project_name {
|
|||||||
VERSION="$(echo "${FULL_TAG}" | cut -d '-' -f 1)"
|
VERSION="$(echo "${FULL_TAG}" | cut -d '-' -f 1)"
|
||||||
|
|
||||||
TMP_DIR="$(mktemp -d)"
|
TMP_DIR="$(mktemp -d)"
|
||||||
cp "${WHL_PATH}" "${TMP_DIR}"
|
wheel unpack "${WHL_PATH}" -d "${TMP_DIR}"
|
||||||
pushd "${TMP_DIR}"
|
TMP_UNPACKED_DIR="$(ls -d "${TMP_DIR}"/* | head -n 1)"
|
||||||
unzip -q "${ORIGINAL_WHL_NAME}"
|
pushd "${TMP_UNPACKED_DIR}"
|
||||||
|
|
||||||
ORIGINAL_WHL_DIR_PREFIX="${ORIGINAL_PROJECT_NAME}-${VERSION}"
|
ORIGINAL_WHL_DIR_PREFIX="${ORIGINAL_PROJECT_NAME}-${VERSION}"
|
||||||
NEW_WHL_DIR_PREFIX="${NEW_PROJECT_NAME}-${VERSION}"
|
NEW_WHL_DIR_PREFIX="${NEW_PROJECT_NAME}-${VERSION}"
|
||||||
mv "${ORIGINAL_WHL_DIR_PREFIX}.dist-info" "${NEW_WHL_DIR_PREFIX}.dist-info"
|
mv "${ORIGINAL_WHL_DIR_PREFIX}.dist-info" "${NEW_WHL_DIR_PREFIX}.dist-info"
|
||||||
mv "${ORIGINAL_WHL_DIR_PREFIX}.data" "${NEW_WHL_DIR_PREFIX}.data" || echo
|
if [[ -d "${ORIGINAL_WHL_DIR_PREFIX}.data" ]]; then
|
||||||
sed -i.bak "s/${ORIGINAL_PROJECT_NAME}/${NEW_PROJECT_NAME}/g" "${NEW_WHL_DIR_PREFIX}.dist-info/RECORD"
|
mv "${ORIGINAL_WHL_DIR_PREFIX}.data" "${NEW_WHL_DIR_PREFIX}.data"
|
||||||
|
fi
|
||||||
|
|
||||||
ORIGINAL_PROJECT_NAME_DASH="${ORIGINAL_PROJECT_NAME//_/-}"
|
ORIGINAL_PROJECT_NAME_DASH="${ORIGINAL_PROJECT_NAME//_/-}"
|
||||||
NEW_PROJECT_NAME_DASH="${NEW_PROJECT_NAME//_/-}"
|
NEW_PROJECT_NAME_DASH="${NEW_PROJECT_NAME//_/-}"
|
||||||
sed -i.bak "s/${ORIGINAL_PROJECT_NAME_DASH}/${NEW_PROJECT_NAME_DASH}/g" "${NEW_WHL_DIR_PREFIX}.dist-info/METADATA"
|
sed -i.bak "s/${ORIGINAL_PROJECT_NAME_DASH}/${NEW_PROJECT_NAME_DASH}/g" "${NEW_WHL_DIR_PREFIX}.dist-info/METADATA"
|
||||||
|
|
||||||
zip -rq "${NEW_WHL_NAME}" "${NEW_WHL_DIR_PREFIX}.dist-info" "${NEW_WHL_DIR_PREFIX}.data" "tensorflow" "tensorflow_core"
|
wheel pack "${TMP_UNPACKED_DIR}" -d "${ORIGINAL_WHL_DIR}"
|
||||||
mv "${NEW_WHL_NAME}" "${ORIGINAL_WHL_DIR}"
|
|
||||||
popd
|
popd
|
||||||
rm -rf "${TMP_DIR}"
|
rm -rf "${TMP_DIR}"
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user