From 60cb891b1cb553deffcd505d58d7b20da1759a25 Mon Sep 17 00:00:00 2001
From: Humberto Gonzalez <hgonzalez@linkedin.com>
Date: Mon, 30 Jan 2023 15:07:54 -0800
Subject: [PATCH] Add sdist tar creation when running setup.py

---
 pip_pkg.sh | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pip_pkg.sh b/pip_pkg.sh
index e19dbbf4cd..cb139b7151 100755
--- a/pip_pkg.sh
+++ b/pip_pkg.sh
@@ -45,7 +45,10 @@ echo >>tensorflow_probability/python/version.py \
 # specifies the output dir) to setup.py, e.g.,
 #  ./pip_pkg /tmp/tensorflow_probability_pkg --gpu --release
 # passes `--gpu --release` to setup.py.
-python setup.py bdist_wheel --universal ${@:2} --dist-dir="$DEST" >/dev/null
+python setup.py ${@:2} \
+  sdist --dist-dir="$DEST" \
+  bdist_wheel --universal --dist-dir="$DEST" \
+  >/dev/null
 
 set +x
 echo -e "\nBuild complete. Wheel files are in $DEST"