The problem about viz_generated_sbp.py

[huzijun1996]

Traceback (most recent call last):
File "viz_generated_sbp.py", line 22, in
import sim_agent
ModuleNotFoundError: No module named 'sim_agent'

Traceback (most recent call last):
File "viz_generated_sbp.py", line 31, in
from transformers.utils import set_seed
ImportError: cannot import name 'set_seed' from 'transformers.utils' (/home/xxx/anaconda3/envs/tip22/lib/python3.8/site-packages/transformers/utils/init.py)

I cannot find sim_agent. I cannot find set_seed from transformers.util. What should I do?

[amaliacontiero]

As mentioned by the author, viz_generated_sbp.py is an uncleaned script and you need to modify it to get it to work.

For starters, these two errors can be fixed by replacing line 22 with "import bullet_agent" and line 31 with "from learning_utils import set_seed". Look at the rest of the python files for useful tools and the commented out code and args of viz_generated_sbp.py to get your desired result. You need to edit accordingly.

[huzijun1996]

Thank you for your answer. My problem has been resolved.