From 2f8f7e216ca561a051295e51c3302024e9f1815a Mon Sep 17 00:00:00 2001
From: Afroz Mohiuddin <afrozm@google.com>
Date: Tue, 26 Oct 2021 13:24:52 -0700
Subject: [PATCH] TRAX v.1.4.1 and updates to the release script for python3.

PiperOrigin-RevId: 405721113
---
 oss_scripts/oss_release.sh | 8 ++++----
 setup.py                   | 3 +--
 2 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/oss_scripts/oss_release.sh b/oss_scripts/oss_release.sh
index 122e74edd..0dee30385 100755
--- a/oss_scripts/oss_release.sh
+++ b/oss_scripts/oss_release.sh
@@ -28,16 +28,16 @@ git clone https://github.com/google/trax.git
 cd trax
 git checkout $GIT_COMMIT_ID
 
-python -m pip install wheel twine pyopenssl
+python3 -m pip install wheel twine pyopenssl
 
 # Build the distribution
 echo "Building distribution"
-python setup.py sdist
-python setup.py bdist_wheel --universal
+python3 setup.py sdist
+python3 setup.py bdist_wheel --universal
 
 # Publish to PyPI
 echo "Publishing to PyPI"
-twine upload dist/*
+python3 -m twine upload dist/*
 
 # Cleanup
 rm -rf build/ dist/ trax.egg-info/
diff --git a/setup.py b/setup.py
index 575f2454b..23f2fc99a 100644
--- a/setup.py
+++ b/setup.py
@@ -13,7 +13,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# coding=utf-8
 # coding=utf-8
 """Install trax."""
 
@@ -22,7 +21,7 @@
 
 setup(
     name='trax',
-    version='1.4.0',
+    version='1.4.1',
     description='Trax',
     long_description=(
         'Trax helps you understand deep learning. We start with basic maths and'