From 7da93f819c4d5f52ce5ccfef0f39a7f836cb0d08 Mon Sep 17 00:00:00 2001
From: Mihir Patel <mihir.v.patel7@gmail.com>
Date: Thu, 4 May 2023 20:35:07 -0700
Subject: [PATCH] Pin protobuf (#2203)

* pin

* <
---
 setup.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index c1dad50199..8fa479f687 100644
--- a/setup.py
+++ b/setup.py
@@ -182,7 +182,10 @@ def package_files(prefix: str, directory: str, extension: str):
     'datasets>=2.4,<3',
 ]
 
-extra_deps['sentencepiece'] = ['sentencepiece==0.1.98']
+extra_deps['sentencepiece'] = [
+    'protobuf<3.21',
+    'sentencepiece==0.1.98',
+]
 
 extra_deps['mlperf'] = [
     # TODO: use pip when available: https://github.com/mlcommons/logging/issues/218