Running dp train leads to issues with tensorflow

[kimiaGF]

I'm not sure if this is an issue with tensorflow dependencies but I'm getting the following error with I run dp train input.json:

Traceback (most recent call last):
  File "/home/kimia.gh/.conda/envs/deepmd/lib/python3.9/site-packages/deepmd/env.py", line 18, in <module>
    import tensorflow.compat.v1 as tf
  File "/home/kimia.gh/.conda/envs/deepmd/lib/python3.9/site-packages/tensorflow/__init__.py", line 38, in <module>
    import six as _six
ModuleNotFoundError: No module named 'six'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/kimia.gh/.conda/envs/deepmd/bin/dp", line 7, in <module>
    from deepmd.entrypoints.main import main
  File "/home/kimia.gh/.conda/envs/deepmd/lib/python3.9/site-packages/deepmd/__init__.py", line 7, in <module>
    import deepmd.utils.network as network
  File "/home/kimia.gh/.conda/envs/deepmd/lib/python3.9/site-packages/deepmd/utils/__init__.py", line 2, in <module>
    from .data import DeepmdData
  File "/home/kimia.gh/.conda/envs/deepmd/lib/python3.9/site-packages/deepmd/utils/data.py", line 10, in <module>
    from deepmd.env import GLOBAL_NP_FLOAT_PRECISION
  File "/home/kimia.gh/.conda/envs/deepmd/lib/python3.9/site-packages/deepmd/env.py", line 22, in <module>
    import tensorflow as tf
  File "/home/kimia.gh/.conda/envs/deepmd/lib/python3.9/site-packages/tensorflow/__init__.py", line 38, in <module>
    import six as _six
ModuleNotFoundError: No module named 'six'

UPDATE seems to be an environment issue since I can run the script in another environment, please disregard.

[likefallwind]

Maybe you can write what environment leads to this problem, to help other people have the same problem

[kimiaGF]

Unfortunately I had already removed the environment, but if I run into the same issue again ill post it!