diff --git a/python/packaging/classic/setup.py b/python/packaging/classic/setup.py index 8844d225d1b8..ddd2448e1c18 100755 --- a/python/packaging/classic/setup.py +++ b/python/packaging/classic/setup.py @@ -275,6 +275,7 @@ def run(self): "pyspark.sql.connect.client", "pyspark.sql.connect.functions", "pyspark.sql.connect.proto", + "pyspark.sql.connect.protobuf", "pyspark.sql.connect.shell", "pyspark.sql.connect.streaming", "pyspark.sql.connect.streaming.worker", diff --git a/python/packaging/connect/setup.py b/python/packaging/connect/setup.py index 157bc86c192e..01c5518d4451 100755 --- a/python/packaging/connect/setup.py +++ b/python/packaging/connect/setup.py @@ -103,6 +103,7 @@ "pyspark.sql.connect.client", "pyspark.sql.connect.functions", "pyspark.sql.connect.proto", + "pyspark.sql.connect.protobuf", "pyspark.sql.connect.shell", "pyspark.sql.connect.streaming", "pyspark.sql.connect.streaming.worker",