Refactoring to 1.5.1

.github/workflows/build.yaml

@@ -21,9 +21,9 @@ name: build
 on:
   # Triggers the workflow on push or pull request events but only for the master branch
   push:
-    branches: [ master ]
+    branches: [ "master" ]
   pull_request:
-    branches: [ master ]
+    branches: [ "master" ]
 
   # Allows you to run this workflow manually from the Actions tab
   workflow_dispatch:
@@ -37,55 +37,97 @@ jobs:
 
     strategy:
       matrix:
-        python-version: [3.7]
-        # tf-nightly has some pip version conflicts, so can't be installed.
-        # Use only numbered TF as of now.
-        # tf-version: ["2.4.*", "tf-nightly"]
-        tf-version: ["2.4.*"]
+        python-version: [ '3.10' ]
+        # Which tf-version run.
+        tf-version: [ '2.13.0' ]
         # Which set of tests to run.
-        trax-test: ["lib", "research"]
+        trax-test: [ 'lib','research' ]
 
     # Steps represent a sequence of tasks that will be executed as part of the job
     steps:
-    # Checks-out your repository under $GITHUB_WORKSPACE, so your job can access it
-    - uses: actions/checkout@v2
-    - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v2
-      with:
-        python-version: ${{ matrix.python-version }}
-    - name: Install dependencies
-      run: |
-        python -m pip install --upgrade pip
-        python -m pip install -q -U setuptools numpy
-        python -m pip install flake8 pytest
-        if [[ ${{matrix.tf-version}} == "tf-nightly" ]]; then python -m pip install tf-nightly; else python -m pip install -q "tensorflow=="${{matrix.tf-version}}; fi
-        pip install -e .[tests,t5]
-    # # Lint with flake8
-    # - name: Lint with flake8
-    #   run: |
-    #     # stop the build if there are Python syntax errors or undefined names
-    #     flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
-    #     # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
-    #     flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
-    # Test out right now with only testing one directory.
-    - name: Test with pytest
-      run: |
-        TRAX_TEST=" ${{matrix.trax-test}}" ./oss_scripts/oss_tests.sh
-    # The below step just reports the success or failure of tests as a "commit status".
-    # This is needed for copybara integration.
-    - name: Report success or failure as github status
-      if: always()
-      shell: bash
-      run: |
-        status="${{ job.status }}"
-        lowercase_status=$(echo $status | tr '[:upper:]' '[:lower:]')
-        curl -sS --request POST \
-        --url https://api.github.com/repos/${{ github.repository }}/statuses/${{ github.sha }} \
-        --header 'authorization: Bearer ${{ secrets.GITHUB_TOKEN }}' \
-        --header 'content-type: application/json' \
-        --data '{
-            "state": "'$lowercase_status'",
-            "target_url": "https://github.com/${{ github.repository }}/actions/runs/${{ github.run_id }}",
-            "description": "'$status'",
-            "context": "github-actions/build"
-            }'
+      # Checks-out your repository under $GITHUB_WORKSPACE, so your job can access it
+      - uses: actions/checkout@v3
+      - name: Set up Python ${{matrix.python-version}}
+        uses: actions/setup-python@v3
+        with:
+          python-version: ${{matrix.python-version}}
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          python -m pip install setuptools 
+          python -m pip install numpy==1.23.5
+          python -m pip install flake8 pytest
+          python -m pip install orbax-checkpoint orbax-export
+          python -m pip install tensorflow_datasets==4.2.0
+          python -m pip install tensorflow_metadata==1.6.0
+          python -m pip install tensorflow-text==2.13.0
+          python -m pip install pandas==2.1.1
+          python -m pip install matplotlib==3.8.0
+          python -m pip install multimethod==1.10
+          python -m pip install natsort==8.4.0
+          python -m pip install omegaconf==2.3.0
+          python -m pip install nltk==3.8.1
+          python -m pip install pytest==7.4.2
+          python -m pip install hydra-core==1.3.2
+          python -m pip install scikit-learn==1.3.1
+          python -m pip install pympler==1.0.1
+          python -m pip install IPython==8.16.1
+          python -m pip install mypy==1.5.1
+          python -m pip install pylint==2.17.7
+          python -m pip install black==23.9.1
+          python -m pip install openpyxl==3.1.2
+          python -m pip install numba==0.58.0
+          python -m pip install parameterized==0.9.0
+          python -m pip install mock==5.1.0
+          python -m pip install tfds-nightly==4.6.0.dev202210050045
+          python -m pip install editdistance==0.6.2
+          python -m pip install pyglove==0.4.3
+          python -m pip install sentencepiece==0.1.99
+          python -m pip install babel==2.13.0
+          python -m pip install rouge-score==0.1.2
+          python -m pip install sacrebleu==2.3.1
+          python -m pip install transformers==4.33.3
+          python -m pip install mesh-tensorflow==0.1.21
+          python -m pip install torch==2.0.1+cu118 -f https://download.pytorch.org/whl/cu118/torch_stable.html
+          python -m pip install protobuf==3.20.3
+          python -m pip install t5==0.9.2 --no-dependencies tensorflow
+          python -m pip install seqio --no-deps tensorflow
+          python -m pip install funcsigs==1.0.2
+          python -m pip install absl-py==1.4.0
+          python -m pip install gym==0.26.2
+          python -m pip install gin-config==0.5.0
+          python -m pip install jax==0.4.20
+          python -m pip install jaxlib==0.4.20
+          python -m pip install psutil==5.9.5
+          python -m pip install scipy==1.11.3
+          python -m pip install six==1.14.0
+          python -m pip install attrs==23.1.0
+          python -m pip install mock==5.1.0
+          python -m pip install parameterized==0.9.0
+          python -m pip install pylint==2.17.7
+          python -m pip install pytest==7.4.2
+          python -m pip install wrapt==1.15.0
+          python -m pip install tensor2tensor==1.15.7
+          python -m pip install clu==0.0.10
+      # Test out right now with only testing one directory.
+      - name: Test with pytest
+        run: |
+          TRAX_TEST="${{matrix.trax-test}}" ./oss_scripts/oss_tests.sh
+      # The below step just reports the success or failure of tests as a "commit status".
+      # This is needed for copy bara integration.
+      - name: Report success or failure as github status
+        if: always()
+        shell: bash
+        run: |
+          status="${{ job.status }}"
+          lowercase_status=$(echo $status | tr '[:upper:]' '[:lower:]')
+          curl -sS --request POST \
+          --url https://api.github.com/repos/${{github.repository}}/statuses/${{github.sha}} \
+          --header 'authorization: Bearer ${{ secrets.GITHUB_TOKEN }}' \
+          --header 'content-type: application/json' \
+          --data '{
+              "state": "'$lowercase_status'",
+              "target_url": "https://github.com/${{github.repository}}/actions/runs/${{github.run_id}}",
+              "description": "'$status'",
+              "context": "github-actions/build"
+              }'

.github/workflows/codeql.yml

@@ -0,0 +1,82 @@
+# For most projects, this workflow file will not need changing; you simply need
+# to commit it to your repository.
+#
+# You may wish to alter this file to override the set of languages analyzed,
+# or to provide custom queries or build logic.
+#
+# ******** NOTE ********
+# We have attempted to detect the languages in your repository. Please check
+# the `language` matrix defined below to confirm you have the correct set of
+# supported CodeQL languages.
+#
+name: "CodeQL"
+
+on:
+  push:
+    branches: [ "1.5.1" ]
+  pull_request:
+    # The branches below must be a subset of the branches above
+    branches: [ "1.5.1" ]
+  schedule:
+    - cron: '31 4 * * 1'
+
+jobs:
+  analyze:
+    name: Analyze
+    # Runner size impacts CodeQL analysis time. To learn more, please see:
+    #   - https://gh.io/recommended-hardware-resources-for-running-codeql
+    #   - https://gh.io/supported-runners-and-hardware-resources
+    #   - https://gh.io/using-larger-runners
+    # Consider using larger runners for possible analysis time improvements.
+    runs-on: ${{ (matrix.language == 'swift' && 'macos-latest') || 'ubuntu-latest' }}
+    timeout-minutes: ${{ (matrix.language == 'swift' && 120) || 360 }}
+    permissions:
+      actions: read
+      contents: read
+      security-events: write
+
+    strategy:
+      fail-fast: false
+      matrix:
+        language: [ 'python' ]
+        # CodeQL supports [ 'c-cpp', 'csharp', 'go', 'java-kotlin', 'javascript-typescript', 'python', 'ruby', 'swift' ]
+        # Use only 'java-kotlin' to analyze code written in Java, Kotlin or both
+        # Use only 'javascript-typescript' to analyze code written in JavaScript, TypeScript or both
+        # Learn more about CodeQL language support at https://aka.ms/codeql-docs/language-support
+
+    steps:
+    - name: Checkout repository
+      uses: actions/checkout@v3
+
+    # Initializes the CodeQL tools for scanning.
+    - name: Initialize CodeQL
+      uses: github/codeql-action/init@v2
+      with:
+        languages: ${{ matrix.language }}
+        # If you wish to specify custom queries, you can do so here or in a config file.
+        # By default, queries listed here will override any specified in a config file.
+        # Prefix the list here with "+" to use these queries and those in the config file.
+
+        # For more details on CodeQL's query packs, refer to: https://docs.github.com/en/code-security/code-scanning/automatically-scanning-your-code-for-vulnerabilities-and-errors/configuring-code-scanning#using-queries-in-ql-packs
+        # queries: security-extended,security-and-quality
+
+
+    # Autobuild attempts to build any compiled languages (C/C++, C#, Go, Java, or Swift).
+    # If this step fails, then you should remove it and run the build manually (see below)
+    - name: Autobuild
+      uses: github/codeql-action/autobuild@v2
+
+    # ℹ️ Command-line programs to run using the OS shell.
+    # 📚 See https://docs.github.com/en/actions/using-workflows/workflow-syntax-for-github-actions#jobsjob_idstepsrun
+
+    #   If the Autobuild fails above, remove it and uncomment the following three lines.
+    #   modify them (or add more) to build your code if your project, please refer to the EXAMPLE below for guidance.
+
+    # - run: |
+    #     echo "Run, Build Application using script"
+    #     ./location_of_script_within_repo/buildscript.sh
+
+    - name: Perform CodeQL Analysis
+      uses: github/codeql-action/analyze@v2
+      with:
+        category: "/language:${{matrix.language}}"

.readthedocs.yaml

@@ -28,6 +28,6 @@ formats: all
 
 # Optionally set the version of Python and requirements required to build your docs
 python:
-  version: 3.7
+  version: 3.10
   install:
     - requirements: docs/requirements.txt

.travis.yml

@@ -5,10 +5,10 @@ git:
   depth: 3
   quiet: true
 python:
-  - "3.6"
+  - "3.10"
 env:
   global:
-    - TF_VERSION="2.4.*"
+    - TF_VERSION="2.11.0"
   matrix:
     - TRAX_TEST="lib"
     - TRAX_TEST="research"

README.md

@@ -5,7 +5,7 @@
 version](https://badge.fury.io/py/trax.svg)](https://badge.fury.io/py/trax)
 [![GitHub
 Issues](https://img.shields.io/github/issues/google/trax.svg)](https://github.com/google/trax/issues)
-![GitHub Build](https://github.com/google/trax/actions/workflows/build.yaml/badge.svg)
+![GitHub Build](https://github.com/mmarcinmichal/trax/actions/workflows/build.yaml/badge.svg)
 [![Contributions
 welcome](https://img.shields.io/badge/contributions-welcome-brightgreen.svg)](CONTRIBUTING.md)
 [![License](https://img.shields.io/badge/License-Apache%202.0-brightgreen.svg)](https://opensource.org/licenses/Apache-2.0)

docs/.readthedocs.yaml

@@ -28,6 +28,6 @@ formats: all
 
 # Optionally set the version of Python and requirements required to build your docs
 python:
-  version: 3.7
+  version: 3.10
   install:
     - requirements: docs/requirements.txt

oss_scripts/oss_pip_install.sh

@@ -15,7 +15,7 @@
 #!/bin/bash
 
 set -v  # print commands as they're executed
-set -e  # fail and exit on any command erroring
+set -e  # fail and exit on any command error
 
 : "${TF_VERSION:?}"
 

oss_scripts/oss_release.sh

@@ -15,18 +15,18 @@
 #!/bin/bash
 
 set -v  # print commands as they're executed
-set -e  # fail and exit on any command erroring
+set -e  # fail and exit on any command error
 
 GIT_COMMIT_ID=${1:-""}
 [[ -z $GIT_COMMIT_ID ]] && echo "Must provide a commit" && exit 1
 
 TMP_DIR=$(mktemp -d)
-pushd $TMP_DIR
+pushd "$TMP_DIR"
 
 echo "Cloning trax and checking out commit $GIT_COMMIT_ID"
 git clone https://github.com/google/trax.git
 cd trax
-git checkout $GIT_COMMIT_ID
+git checkout "$GIT_COMMIT_ID"
 
 python3 -m pip install wheel twine pyopenssl
 
@@ -42,4 +42,4 @@ python3 -m twine upload dist/*
 # Cleanup
 rm -rf build/ dist/ trax.egg-info/
 popd
-rm -rf $TMP_DIR
+rm -rf "$TMP_DIR"