AttributeError: module 'trax.supervised.trainer_lib' has no attribute 'init_random_number_generators'

[topekekere]

Description

Attribute Error

Environment information

Google Colab

OS: Ubuntu 20.04.3 LTS

$ pip freeze | grep trax
trax==1.4.1

$ pip freeze | grep tensor
tensorboard==2.6.0
tensorboard-data-server==0.6.1
tensorboard-plugin-wit==1.8.0
tensorflow @ file:///tensorflow-2.6.0-cp37-cp37m-linux_x86_64.whl
tensorflow-datasets==4.0.1
tensorflow-estimator==2.6.0
tensorflow-gcs-config==2.6.0
tensorflow-hub==0.12.0
tensorflow-metadata==1.2.0
tensorflow-probability==0.14.1
tensorflow-text==2.6.0

$ pip freeze | grep jax
jax==0.2.21
jaxlib @ https://storage.googleapis.com/jax-releases/cuda111/jaxlib-0.1.71+cuda111-cp37-none-manylinux2010_x86_64.whl

$ python -V
Python 3.7.12

For bugs: reproduction and error logs

---

AttributeError Traceback (most recent call last)

in () 6 7 # set random seeds to make this notebook easier to replicate ----> 8 trax.supervised.trainer_lib.init_random_number_generators(31) 9 10 # import trax.fastmath.numpy AttributeError: module 'trax.supervised.trainer_lib' has no attribute 'init_random_number_generators' ``` # Steps to reproduce: ``` import os import random as rnd import trax trax.supervised.trainer_lib.init_random_number_generators(31) import trax.fastmath.numpy as np from trax import layers as tl from utils import Layer, load_tweets, process_tweet ``` ``` # Error logs: AttributeError: module 'trax.supervised.trainer_lib' has no attribute 'init_random_number_generators' ``` 

[topekekere]

Untitled19 (1).pdf