Exclude benchmarks directory from distribution (#19861)

* Exclude benchmarks directory from distribution * lint * format with bash shell/format.sh
2024-06-16 15:08:46 -04:00 · 2024-06-16 15:08:46 -04:00 · 45e1175b78
commit 45e1175b78
parent 727b7ae4e1
1 changed files with 4 additions and 1 deletions
--- 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"),
+    ),
 )