From 45e1175b785514593092ba0b197a562df512017b Mon Sep 17 00:00:00 2001
From: Mark Harfouche <mark.harfouche@gmail.com>
Date: Sun, 16 Jun 2024 15:08:46 -0400
Subject: [PATCH] Exclude benchmarks directory from distribution (#19861)

* Exclude benchmarks directory from distribution

* lint

* format with bash shell/format.sh
---
 setup.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 822eccfa7..6d8096a0b 100644
--- a/setup.py
+++ b/setup.py
@@ -60,5 +60,8 @@ setup(
         "Topic :: Scientific/Engineering",
         "Topic :: Software Development",
     ],
-    packages=find_packages(exclude=("*_test.py",)),
+    packages=find_packages(
+        include=("keras", "keras.*"),
+        exclude=("*_test.py", "benchmarks"),
+    ),
 )