From b2099049484f66f4348ddd4448c676feecb0b46e Mon Sep 17 00:00:00 2001
From: Bu Sun Kim <8822365+busunkim96@users.noreply.github.com>
Date: Tue, 15 Jun 2021 18:52:14 -0600
Subject: [PATCH] fix: exclude docs and tests from package (#481)

Only include packages that start with google in the published artifact
---
 setup.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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"