update release scripts

PiperOrigin-RevId: 488513372

praxis/pip_package/cloudbuild-release.yaml

@@ -6,7 +6,7 @@ steps:
           '-f', 'praxis/pip_package/release.Dockerfile', '.',
           '--build-arg', 'wheel_folder=${_WHEEL_FOLDER}',
         ]
-  timeout: 14400s
+  timeout: 3600s
 - name: 'gcr.io/cloud-builders/docker'
   args: ['push', '--all-tags', 'gcr.io/${PROJECT_ID}/${_IMAGE_NAME}']
   timeout: 1800s
@@ -22,8 +22,8 @@ substitutions:
 options:
     dynamic_substitutions: true
     substitution_option: 'ALLOW_LOOSE'
-timeout: 24000s
+timeout: 5400s
 artifacts:
   objects:
     location: 'gs://pax-on-cloud-tpu-project/wheels/$(date -u +%Y%m%d)-praxis-${_RELEASE_VERSION}'
-    paths: ['/**/*.whl']
+    paths: ['/**/*.whl']

praxis/pip_package/release.Dockerfile

@@ -16,25 +16,12 @@ ENV PYTHON_MINOR_VERSION="8"
 # Pick up some TF dependencies
 RUN apt update && DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends software-properties-common
 RUN apt update && DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends \
-        aria2 \
         build-essential \
         curl \
-        dirmngr \
-        emacs \
         git \
-        gpg-agent \
-        less \
-        libfreetype6-dev \
-        libhdf5-serial-dev \
-        libpng-dev \
-        libzmq3-dev \
-        lsof \
         pkg-config \
         rename \
         rsync \
-        python-dev \
-        python3-distutils \
-        sox \
         unzip \
         vim \
         && \
@@ -66,7 +53,7 @@ RUN mkdir /bazel && \
 
 COPY . /praxis
 RUN mkdir $WHEEL_FOLDER
-RUN pip3 install /praxis/praxis/pip_package
+RUN pip3 install -r /praxis/praxis/pip_package/requirements.txt
 
 RUN git clone https://github.com/google/flaxformer.git
 RUN cd flaxformer && pip3 install .