Not able to import trax in google colab though trax is installed

[ankusri]

Description

I am trying trax-ml library in Google Colab.
I installed trax using below command
!pip install trax==1.3.1

But I am not able to import trax library.

Please help me in this issue.

Environment information

Google Colab

OS: <your answer here>

$ pip freeze | grep trax
# your output here

$ pip freeze | grep tensor
# your output here

$ pip freeze | grep jax
# your output here

$ python -V
# your output here

For bugs: reproduction and error logs

# Steps to reproduce:
...!pip install trax==1.3.1
import trax

# Error logs:
...
```---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
[<ipython-input-3-3afdcd9c7c68>](https://localhost:8080/#) in <module>()
----> 1 import trax

2 frames
[/usr/local/lib/python3.7/dist-packages/trax/layers/__init__.py](https://localhost:8080/#) in layer_configure(*args, **kwargs)
     41 def layer_configure(*args, **kwargs):
     42   kwargs['module'] = 'trax.layers'
---> 43   return gin.external_configurable(*args, **kwargs)
     44 
     45 # pylint: disable=used-before-assignment

TypeError: external_configurable() got an unexpected keyword argument 'blacklist'

[ngctnnnn]

You could try with

!pip install -q -U git+https://github.com/google/trax@master