From 766bcf5d002fd63c198b7ca9b6a4093462b08783 Mon Sep 17 00:00:00 2001
From: Kalyani Nikure <knikure@amazon.com>
Date: Thu, 18 May 2023 17:37:23 +0000
Subject: [PATCH 1/2] Bump to protobuf4 and remove unused dependency
 protobuf3-to-dict

---
 setup.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index e313587bec..2c48d65d74 100644
--- a/setup.py
+++ b/setup.py
@@ -52,8 +52,7 @@ def read_requirements(filename):
     "cloudpickle==2.2.1",
     "google-pasta",
     "numpy>=1.9.0,<2.0",
-    "protobuf>=3.1,<4.0",
-    "protobuf3-to-dict>=0.1.5,<1.0",
+    "protobuf>=4.21.1,<5.0",
     "smdebug_rulesconfig==1.0.1",
     "importlib-metadata>=1.4.0,<5.0",
     "packaging>=20.0",

From 232543b7c4d643c39bf2697184dae7430437e919 Mon Sep 17 00:00:00 2001
From: Kalyani Nikure <knikure@amazon.com>
Date: Fri, 19 May 2023 22:13:26 +0000
Subject: [PATCH 2/2] Install other google packages to extend support for
 protobuf4

---
 setup.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/setup.py b/setup.py
index 2c48d65d74..3d0f781a60 100644
--- a/setup.py
+++ b/setup.py
@@ -51,6 +51,12 @@ def read_requirements(filename):
     "boto3>=1.26.131,<2.0",
     "cloudpickle==2.2.1",
     "google-pasta",
+    "google-api-core==2.8.0",
+    "google-auth==2.6.6",
+    "google-cloud-core==2.3.0",
+    "google-cloud-storage==2.3.0",
+    "google-resumable-media==2.3.3",
+    "googleapis-common-protos==1.56.1",
     "numpy>=1.9.0,<2.0",
     "protobuf>=4.21.1,<5.0",
     "smdebug_rulesconfig==1.0.1",