search

rail-berkeley / serl

SERL: A Software Suite for Sample-Efficient Robotic Reinforcement Learning
https://serl-robot.github.io/
MIT License
375 stars 42 forks source link

No module named 'jaxrl_m.data.' #43

Closed elishafer closed 6 months ago

elishafer commented 6 months ago

I recently got back to working on serl after a 2 week break and now I'm getting the following message when trying to run bc training bash run_bc.sh. The script worked previously. Has anyone encountered this issue?

  File "/home/user/serl/serl_launcher/serl_launcher/utils/launcher.py", line 10, in <module>
    from agentlace.data.tfds import populate_datastore
  File "/home/user/miniconda3/envs/serl/lib/python3.10/site-packages/agentlace/data/tfds.py", line 8, in <module>
    from agentlace.data.jaxrl_data_store import ReplayBufferDataStore
  File "/home/user/miniconda3/envs/serl/lib/python3.10/site-packages/agentlace/data/jaxrl_data_store.py", line 17, in <module>
    from jaxrl_m.data.replay_buffer import ReplayBuffer
ModuleNotFoundError: No module named 'jaxrl_m.data'
jianlanluo commented 6 months ago

@youliangtan @charlesxu0124 this is probably python path issue again? Could we actually solve it fundamentally?

fwiw, you can try run the script with python -m

youliangtan commented 6 months ago

Previous agentlace was using an internal repo we have. updated the commit in the dep tree to use the lastest fixed version of agentlace.

elishafer commented 6 months ago

Before you responded I created a new conda env to solve the issue so I haven't checked the solution. I'll update if I get to check it or if the issue crops up again edit: wording