ModuleNotFoundError: No module named 'robomimic'

[HJX-exoskeleton]

error:
Traceback (most recent call last):
File "/home/song/hjx/aloha/mobile-aloha_multi-sensor/act-plus-plus/imitate_episodes.py", line 20, in
from policy import ACTPolicy, CNNMLPPolicy, DiffusionPolicy
File "/home/song/hjx/aloha/mobile-aloha_multi-sensor/act-plus-plus/policy.py", line 11, in
from robomimic.models.base_nets import ResNet18Conv, SpatialSoftmax
ModuleNotFoundError: No module named 'robomimic'

[TangGuohh]

look this :
also need to install https://github.com/ARISE-Initiative/robomimic/tree/r2d2 (note the r2d2 branch) for Diffusion Policy by pip install -e .

you need to git clone https://github.com/ARISE-Initiative/robomimic/tree/r2d2
and
Run pip install -e . in robomimic
Run pip install -e . in robomimic/act/detr

[TangGuohh]

small black son