Sonnet issue with tf2

[suhani247]

Hi, I'm trying to run AndroidEnv script Run Acme agent - https://github.com/deepmind/android_env/blob/main/examples/run_acme_agent.py

When I try to run this, after installing acme, I get issue with sonnet which is trying to use tf.contrib.eager.python which is available only for tf1, whereas when I downgrade tensorflow to 1 (and correspondingly python to 2.7(, acme setup.py fails with syntax error.
What is the resolution here?

OS - Ubuntu 22.04