diff --git a/setup.py b/setup.py index d1cced2675..8eaa9454c4 100644 --- a/setup.py +++ b/setup.py @@ -44,7 +44,11 @@ version=version, description=description, long_description=readme, - packages=setuptools.PEP420PackageFinder.find(), + packages=[ + package + for package in setuptools.PEP420PackageFinder.find() + if package.startswith("google") + ], entry_points={ "console_scripts": [ "tb-gcp-uploader=google.cloud.aiplatform.tensorboard.uploader_main:run_main"