ImportError: cannot import name 'ArmActionMode'

[Sid-B-K]

I am getting this issue (in the screenshot) while running RLzoo with RLbench using the following code:

from rlzoo.common.env_wrappers import *
from rlzoo.common.utils import *
from rlzoo.algorithms import *

EnvName = 'ReachTarget'
EnvType = 'rlbench'
env = build_env(EnvName, EnvType, state_type='vision')

AlgName = 'SAC'
alg_params, learn_params = call_default_params(env, EnvType, AlgName)
alg = eval(AlgName+'(**alg_params)')
alg.learn(env=env, mode='train', render=False, **learn_params)
alg.learn(env=env, mode='test', render=True, **learn_params)

env.close()

I have also added export PYTHONPATH="/home/sidharth/RLBench" in .bashrc

Any help would be appreciated! Thanks.

[Sid-B-K]

This issue is now fixed!

The problem was in RLzoo/rlzoo/common/build_rlbench_env.py, there is now a new action mode API in RLBench that allows for easier custom action modes.

The code in this file should be modified, the commented code should be replaced by the code below (as shown in the screenshots).

After editing, go to the RLzoo folder, using cd RLzoo and type pip3 install . to rebuild it.