Ray: [rllib] ModuleNotFoundError: No module named 'tree'

[forhonourlx]

Ray version and other system information (Python version, TensorFlow version, OS):
Python3.6
pip install -U ray-0.9.0.dev0-cp36-cp36m-manylinux1_x86_64.whl
On 2020-03-18

Reproduction (REQUIRED)

$ rllib train -f atari-ddppo.yaml

Traceback (most recent call last):
File "/home/simon/anaconda3/bin/rllib", line 5, in
from ray.rllib.scripts import cli
File "/home/simon/anaconda3/lib/python3.6/site-packages/ray/rllib/__init__.py", line 9, in
from ray.rllib.evaluation.policy_graph import PolicyGraph
File "/home/simon/anaconda3/lib/python3.6/site-packages/ray/rllib/evaluation/__init__.py", line 2, in
from ray.rllib.evaluation.rollout_worker import RolloutWorker
File "/home/simon/anaconda3/lib/python3.6/site-packages/ray/rllib/evaluation/rollout_worker.py", line 19, in
from ray.rllib.evaluation.sampler import AsyncSampler, SyncSampler
File "/home/simon/anaconda3/lib/python3.6/site-packages/ray/rllib/evaluation/sampler.py", line 11, in
from ray.rllib.evaluation.sample_batch_builder import
File "/home/simon/anaconda3/lib/python3.6/site-packages/ray/rllib/evaluation/sample_batch_builder.py", line 6, in
from ray.rllib.policy.sample_batch import SampleBatch, MultiAgentBatch
File "/home/simon/anaconda3/lib/python3.6/site-packages/ray/rllib/policy/__init__.py", line 2, in
from ray.rllib.policy.torch_policy import TorchPolicy
File "/home/simon/anaconda3/lib/python3.6/site-packages/ray/rllib/policy/torch_policy.py", line 10, in
from ray.rllib.utils.torch_ops import convert_to_non_torch_type
File "/home/simon/anaconda3/lib/python3.6/site-packages/ray/rllib/utils/torch_ops.py", line 1, in
import tree
ModuleNotFoundError: No module named 'tree'

[CesMak]

I have the same error.....
Any solution found so far

[sven1977]

For now, could you simply try: pip install dm-tree to fix this?
Let me know, whether this works.

[forhonourlx]

Thanks, it does work.

[sven1977]

Ok, great! Sorry, we added this requirement in one of the recent PRs.

From setup.py: But yeah, if you just update your ray repo without wheel upgrade, this would have failed.

extras["rllib"] = extras["tune"] + [
    "atari_py",
    "dm_tree",
    "gym[atari]",
    "lz4",
    "opencv-python-headless",
    "pyyaml",
    "scipy",
]

[ibrottier]

For now, could you simply try: pip install dm-tree to fix this?
Let me know, whether this works.
@sven1977
Im having this problem when importing "from ray.rllib.env import MultiAgentEnv" and the pip install didn't seem to work

any idea?

[andykat]

I am also still having issue after trying pip install dm-tree

[caesar025]

I am also still having this issue, despite the pip install. Any ideas on how to fix?

[XuandongZhao]

Still have the same error after installing dm-tree. And my python version is python3.8.

[zzchuman]

Still have the same error after installing dm-tree. And my python version is python3.8.

have you solve it? I have the same error on win10 system.

[dciborow]

pip uninstall tree
pip uninstall dm-tree

pip install --upgrade ray
pip install dm-tree

Working now