Update Python wheel build rules

$ make -C tensorflow_lite_support/tools/pip_package/rpi docker-build PYTHON_VERSION="3.10"

PiperOrigin-RevId: 546776675

Author: terryheo

tensorflow_lite_support/tools/pip_package/rpi/Dockerfile.py3

@@ -15,11 +15,13 @@
 ARG IMAGE
 FROM ${IMAGE}
 ARG PYTHON_VERSION
+ARG NUMPY_VERSION
 
 COPY update_sources.sh /
 RUN /update_sources.sh
 
 RUN apt-get update && \
+    DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC \
     apt-get install -y \
       build-essential \
       software-properties-common \
@@ -30,7 +32,6 @@ RUN apt-get update && \
       xxd && \
     apt-get clean
 
-RUN DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt-get -y install tzdata
 # Install Python packages.
 RUN dpkg --add-architecture armhf
 RUN dpkg --add-architecture arm64
@@ -53,7 +54,7 @@ RUN curl -OL https://bootstrap.pypa.io/get-pip.py
 RUN python3 get-pip.py
 RUN rm get-pip.py
 RUN pip3 install --upgrade pip
-RUN pip3 install numpy~=1.20.0 setuptools pybind11
+RUN pip3 install numpy~=$NUMPY_VERSION setuptools pybind11
 RUN ln -sf /usr/include/python$PYTHON_VERSION /usr/include/python3
 RUN ln -sf /usr/local/lib/python$PYTHON_VERSION/dist-packages/numpy/core/include/numpy /usr/include/python3/numpy
 RUN ln -sf /usr/bin/python3 /usr/bin/python

tensorflow_lite_support/tools/pip_package/rpi/Makefile

@@ -15,6 +15,7 @@
 # Values: debian:<version>, ubuntu:<version>
 BASE_IMAGE ?= ubuntu:20.04
 PYTHON_VERSION ?= 3.9
+NUMPY_VERSION ?= 1.21.2
 
 # Nightly flag for build_arm_pip_package.sh
 NIGHLTY_FLAG ?= --nightly_flag
@@ -46,7 +47,11 @@ help:
 	@echo "make clean        -- remove wheel and deb files"
 
 docker-image:
-	docker build -t $(TAG_IMAGE) --build-arg IMAGE=$(BASE_IMAGE) --build-arg PYTHON_VERSION=$(PYTHON_VERSION) -f $(MAKEFILE_DIR)/Dockerfile.py3 $(MAKEFILE_DIR)/.
+	docker build -t $(TAG_IMAGE) \
+		--build-arg IMAGE=$(BASE_IMAGE) \
+		--build-arg PYTHON_VERSION=$(PYTHON_VERSION) \
+		--build-arg NUMPY_VERSION=$(NUMPY_VERSION) \
+		-f $(MAKEFILE_DIR)/Dockerfile.py3 $(MAKEFILE_DIR)/.
 
 docker-shell: docker-image
 	mkdir -p $(WORKSPACE_DIR)/bazel-ci_build-cache