Trax Import Error

[memora0101]

Description

Hello, I am having issues with the tensorflow imports while using trax, I've never experienced the issue before and not sure how to solve it.

Environment information

---

NotFoundError Traceback (most recent call last) in 11 from nltk.tokenize import sent_tokenize, word_tokenize 12 from nltk.stem import WordNetLemmatizer ---> 13 import trax 14 import trax.fastmath.numpy as np 15 from trax import layers as tl

~/opt/anaconda3/lib/python3.7/site-packages/trax/init.py in 16 """Trax top level import.""" 17 ---> 18 from trax import data 19 from trax import fastmath 20 from trax import layers

~/opt/anaconda3/lib/python3.7/site-packages/trax/data/init.py in 67 from trax.data.inputs import UniformlySeek 68 ---> 69 from trax.data.tf_inputs import add_eos_to_output_features 70 from trax.data.tf_inputs import BertGlueEvalStream 71 from trax.data.tf_inputs import BertGlueTrainStream

~/opt/anaconda3/lib/python3.7/site-packages/trax/data/tf_inputs.py in 30 import numpy as np 31 import scipy ---> 32 import t5.data 33 import tensorflow as tf 34 import tensorflow_datasets as tfds

~/opt/anaconda3/lib/python3.7/site-packages/t5/init.py in 15 """Import API modules.""" 16 ---> 17 import t5.data 18 import t5.evaluation 19 import t5.models

~/opt/anaconda3/lib/python3.7/site-packages/t5/data/init.py in 15 """Import data modules.""" 16 ---> 17 from t5.data.dataset_providers import * # pylint:disable=wildcard-import 18 from t5.data.feature_converters import * # pylint:disable=wildcard-import 19 from t5.data.glue_utils import * # pylint:disable=wildcard-import

~/opt/anaconda3/lib/python3.7/site-packages/t5/data/dataset_providers.py in 30 import dataclasses 31 import gin ---> 32 from t5.data import utils 33 from t5.data.preprocessors import tokenize as tokenize_preprocessor 34 from t5.data.vocabularies import Vocabulary

~/opt/anaconda3/lib/python3.7/site-packages/t5/data/utils.py in 25 import gin 26 import numpy as np ---> 27 from t5.data import vocabularies 28 import tensorflow.compat.v2 as tf 29 import tensorflow_datasets as tfds

~/opt/anaconda3/lib/python3.7/site-packages/t5/data/vocabularies.py in 21 22 import tensorflow.compat.v2 as tf ---> 23 import tensorflow_text as tf_text 24 25 import sentencepiece as sentencepiece_processor

~/opt/anaconda3/lib/python3.7/site-packages/tensorflow_text/init.py in 19 # pylint: disable=wildcard-import 20 from tensorflow_text.python import keras ---> 21 from tensorflow_text.python import metrics 22 from tensorflow_text.python.ops import * 23

~/opt/anaconda3/lib/python3.7/site-packages/tensorflow_text/python/metrics/init.py in 18 19 # pylint: disable=wildcard-import ---> 20 from tensorflow_text.python.metrics.text_similarity_metric_ops import * 21 22 # Public symbols in the "tensorflow_text.metrics" package.

~/opt/anaconda3/lib/python3.7/site-packages/tensorflow_text/python/metrics/text_similarity_metric_ops.py in 26 from tensorflow.python.framework import load_library 27 from tensorflow.python.platform import resource_loader ---> 28 gen_text_similarity_metric_ops = load_library.load_op_library(resource_loader.get_path_to_datafile('_text_similarity_metric_ops.dylib')) 29 30

~/opt/anaconda3/lib/python3.7/site-packages/tensorflow/python/framework/load_library.py in load_op_library(library_filename) 56 RuntimeError: when unable to load the library or get the python wrappers. 57 """ ---> 58 lib_handle = py_tf.TF_LoadLibrary(library_filename) 59 try: 60 wrappers = _pywrap_python_op_gen.GetPythonWrappers(

NotFoundError: dlopen(/Users/lexroverts/opt/anaconda3/lib/python3.7/site-packages/tensorflow_text/python/metrics/_text_similarity_metric_ops.dylib, 6): Symbol not found: __ZN10tensorflow15TensorShapeBaseINS_11TensorShapeEEC2EN4absl14lts_2020_02_254SpanIKxEE Referenced from: /Users/lexroverts/opt/anaconda3/lib/python3.7/site-packages/tensorflow_text/python/metrics/_text_similarity_metric_ops.dylib Expected in: /Users/lexroverts/opt/anaconda3/lib/python3.7/site-packages/tensorflow/python/../libtensorflow_framework.2.dylib in /Users/lexroverts/opt/anaconda3/lib/python3.7/site-packages/tensorflow_text/python/metrics/_text_similarity_metric_ops.dylib

OS: 10.15.7

$ pip freeze | grep trax
# your output here
trax==1.3.9
$ pip freeze | grep tensor
# your output here
mesh-tensorflow==0.1.18
tensor2tensor==1.15.7
tensorboard==2.5.0
tensorboard-data-server==0.6.1
tensorboard-plugin-wit==1.8.0
tensorflow==2.5.0
tensorflow-addons==0.12.1
tensorflow-datasets==4.2.0
tensorflow-estimator==2.5.0
tensorflow-gan==2.0.0
tensorflow-hub==0.11.0
tensorflow-metadata==0.27.0
tensorflow-probability==0.7.0
tensorflow-text==2.4.3
$ pip freeze | grep jax
# your output here
jax==0.2.9
jaxlib==0.1.59
$ python -V
# Python 3.7.6