Seq flow lite does not build

Prerequisites

Please answer the following questions for yourself before submitting an issue.

• [x ] I am using the latest TensorFlow Model Garden release and TensorFlow 2.
  Latest version cloned right from github but seq_flow_lite has TF 2.3 and Python 3.6 as requirements.
• [x ] I am reporting the issue to the correct repository. (Model Garden official or research directory)
• [x ] I checked to make sure that this issue has not already been filed.

1. The entire URL of the file you are using

https://github.com/tensorflow/models/blob/master/research/seq_flow_lite/demo/colab/emotion_colab.ipynb

https://github.com/tensorflow/models/tree/master/research/seq_flow_lite#readme

2. Describe the bug

There are three ways to launch a demo here and none of them work (won't build).

1. Emotion colab demo fails on the line !pip install models/research/seq_flow_lite - maybe because colab uses Python 3.7?
2. From the readme, the command bazel run -c opt :trainer fails to build in a tensorflow/tensorflow:2.3.4-gpu docker image where I installed bazel
3. From the readme, the command bazel run -c opt sgnn:train does not run since it's not defined in the BUILD file - fair enough

3. Steps to reproduce

1. Run the colab.

2. & 3.

git clone https://github.com/tensorflow/models.git
docker run --rm=true -v $(pwd)/models:/home/models -it tensorflow/tensorflow:2.3.4-gpu bash
apt install apt-transport-https curl gnupg
curl -fsSL https://bazel.build/bazel-release.pub.gpg | gpg --dearmor >bazel-archive-keyring.gpg
mv bazel-archive-keyring.gpg /usr/share/keyrings
echo "deb [arch=amd64 signed-by=/usr/share/keyrings/bazel-archive-keyring.gpg] https://storage.googleapis.com/bazel-apt stable jdk1.8" | tee /etc/apt/sources.list.d/bazel.list
apt update 
apt install bazel
cd /home/models/research/seq_flow_lite
bazel run -c opt :trainer -- --config_path=$(pwd)/configs/civil_comments_prado.txt --runner_mode=train --logtostderr --output_dir=/tmp/prado
bazel run -c opt sgnn:train -- --logtostderr --output_dir=/tmp/sgnn

4. Expected behavior

The thing is that I don't even care about PRADO or sgnn. I just wanted to try one of the new models you advertised on the Google AI blog (https://ai.googleblog.com/2022/08/efficient-sequence-modeling-for-on.html) but I can't because it needs the file tf_custom_ops_py.py but it doesn't exist. However tf_custom_ops.cc does exist so it seemed likely that if I could get build the Python version from C++ file. But nothing builds. There are no instructions on how to build and none of the demos seem to work.

5. Additional context

1. Error message from the colab instance:

Building wheels for collected packages: seq-flow-lite
  Building wheel for seq-flow-lite (setup.py) ... error
  ERROR: Failed building wheel for seq-flow-lite
  Running setup.py clean for seq-flow-lite
Failed to build seq-flow-lite
Installing collected packages: seq-flow-lite
    Running setup.py install for seq-flow-lite ... error
ERROR: Command errored out with exit status 1: /usr/bin/python3 -u -c 'import io, os, sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-req-build-cr_5bdz5/setup.py'"'"'; __file__='"'"'/tmp/pip-req-build-cr_5bdz5/setup.py'"'"';f = getattr(tokenize, '"'"'open'"'"', open)(__file__) if os.path.exists(__file__) else io.StringIO('"'"'from setuptools import setup; setup()'"'"');code = f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, __file__, '"'"'exec'"'"'))' install --record /tmp/pip-record-sxy3_k0l/install-record.txt --single-version-externally-managed --compile --install-headers /usr/local/include/python3.7/seq-flow-lite Check the logs for full command output.

6. System information

• OS Platform and Distribution (e.g., Linux Ubuntu 16.04): colab + Ubuntu 18.04.5
• TensorFlow installed from (source or binary): binary
• TensorFlow version (use command below): colab + 2.3.5
• Python version: colab + 3.6.9
• Bazel version (if compiling from source): 5.2.0 + 5.2.0
• GCC/Compiler version (if compiling from source): 7.5.0 + 7.5.0
• CUDA/cuDNN version: 11.1 + 10.1