Open ankusri opened 2 years ago
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.
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
# 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'
You could try with
!pip install -q -U git+https://github.com/google/trax@master
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
For bugs: reproduction and error logs