diff --git a/python/tflite_micro/pypi_build.sh b/python/tflite_micro/pypi_build.sh
index 05812257721..166071f783b 100755
--- a/python/tflite_micro/pypi_build.sh
+++ b/python/tflite_micro/pypi_build.sh
@@ -101,13 +101,15 @@ docker run \
     }
 
     # Build the wheel via bazel, using the Python compatibility tag matching the
-    # build environment.
+    # build environment. Enable compression support for the official package.
     call_bazel build //python/tflite_micro:whl.dist \
-        --//python/tflite_micro:compatibility_tag=\$PY_COMPATIBILITY
+        --//python/tflite_micro:compatibility_tag=\$PY_COMPATIBILITY \
+        --//:with_compression=true
 
     # Test, in the container environment.
     call_bazel test //python/tflite_micro:whl_test \
-            --//python/tflite_micro:compatibility_tag=\$PY_COMPATIBILITY
+            --//python/tflite_micro:compatibility_tag=\$PY_COMPATIBILITY \
+            --//:with_compression=true
 EOF
 
 # Make the output directory tree writable so it can be removed easily by the