Fix arm64 macos build issue #3567

Job	Run time
Build Number	0s
Lint for Docs	15s
Lint	3m 42s
Linux	9m 14s
Bazel Linux	43m 58s
macOS	16m 31s
Bazel macOS	26m 21s
Bazel Windows	32m 5s
Wheel 3.9 macOS	38s
Wheel 3.10 macOS	34s
Wheel 3.11 macOS	39s
Wheel 3.12 macOS	44s
Test 3.9 macOS	1h 3m 57s
Nightly ${{ matrix.python }} macOS	0s
Wheel ${{ matrix.python }} Windows	0s
Test ${{ matrix.python }} Windows	0s
Nightly ${{ matrix.python }} Windows	0s
Wheel 3.9 Linux	1m 13s
Wheel 3.10 Linux	1m 9s
Wheel 3.11 Linux	1m 12s
Wheel 3.12 Linux	1m 12s
Test 3.9 Linux	28m 47s
Release	0s
Nightly ${{ matrix.python }} Linux	0s
Release Candidate	0s
Docker Nightly	0s
Nightly	0s
Docker Release	0s
	3h 52m 11s

Provide feedback