/home/vikisu/alphageometry/lib/python3.10/site-packages/flax/struct.py:132: FutureWarning: jax.tree_util.register_keypaths is deprecated, and will be removed in a future release. Please use register_pytree_with_keys() instead.
jax.tree_util.register_keypaths(data_clz, keypaths)
/home/vikisu/alphageometry/lib/python3.10/site-packages/flax/struct.py:132: FutureWarning: jax.tree_util.register_keypaths is deprecated, and will be removed in a future release. Please use register_pytree_with_keys() instead.
jax.tree_util.register_keypaths(data_clz, keypaths)
/home/vikisu/alphageometry/lib/python3.10/site-packages/flax/struct.py:132: FutureWarning: jax.tree_util.register_keypaths is deprecated, and will be removed in a future release. Please use register_pytree_with_keys() instead.
jax.tree_util.register_keypaths(data_clz, keypaths)
Traceback (most recent call last):
File "/usr/lib/python3.10/runpy.py", line 196, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/usr/lib/python3.10/runpy.py", line 86, in _run_code
exec(code, run_globals)
File "/home/vikisu/alphageometry/alphageometry.py", line 28, in
import lm_inference as lm
File "/home/vikisu/alphageometry/lm_inference.py", line 20, in
import models # pylint: disable=unused-import
File "/home/vikisu/alphageometry/models.py", line 24, in
from transformer import models
File "/home/vikisu/alphageometry/meliad_lib/meliad/transformer/models.py", line 25, in
import metrics_summary
File "/home/vikisu/alphageometry/meliad_lib/meliad/metrics_summary.py", line 20, in
from clu import metric_writers
File "/home/vikisu/alphageometry/lib/python3.10/site-packages/clu/metric_writers/init.py", line 52, in
from clu.metric_writers.summary_writer import SummaryWriter
File "/home/vikisu/alphageometry/lib/python3.10/site-packages/clu/metric_writers/summary_writer.py", line 27, in
import tensorflow as tf
File "/home/vikisu/alphageometry/lib/python3.10/site-packages/tensorflow/init.py", line 38, in
from tensorflow.python.tools import module_util as _module_util
ModuleNotFoundError: No module named 'tensorflow.python'
I had tried to run the command in Ubuntu with python3.10:
then I got the below issue: