Fix bazel build for 51253b3 (#173437)

Commit

25 days ago

Fix bazel build for 51253b3 (#173437) Co-authored-by: Pranav Kant <prka@google.com>