TRAX v.1.4.1 and updates to the release script for python3.

PiperOrigin-RevId: 405721113

Author: afrozenator

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/

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'