From e1eef3c2510cfdc97c67b5b8234e54aac3748414 Mon Sep 17 00:00:00 2001
From: Michael Reneer <michaelreneer@google.com>
Date: Tue, 10 Oct 2023 15:30:21 -0700
Subject: [PATCH] Add a description to the TFP Python package.

PiperOrigin-RevId: 572384355
---
 setup.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index be38b2f6..452f13cc 100644
--- a/setup.py
+++ b/setup.py
@@ -20,11 +20,20 @@
   exec(file.read(), globals_dict)  # pylint: disable=exec-used
   VERSION = globals_dict['__version__']
 
+README = (
+    'A Python library that includes implementations of TensorFlow optimizers '
+    'for training machine learning models with differential privacy.'
+)
+
 setuptools.setup(
     name='tensorflow_privacy',
     version=VERSION,
+    description='A privacy-focused machine learning framework',
+    long_description=README,
+    long_description_content_type='text/plain',
     url='https://github.com/tensorflow/privacy',
     license='Apache-2.0',
+    packages=setuptools.find_packages(),
     install_requires=[
         'absl-py>=1.0,==1.*',
         'attrs>=21.4',
@@ -43,6 +52,5 @@
         'tensorflow~=2.4',
         'tf-models-official~=2.13',
     ],
-    packages=setuptools.find_packages(),
     python_requires='>=3.9.0,<3.12',
 )