I cannot import stable_baselines3 to the script (sofapython3). With the same environment setting by conda, I could import stable_baselines3 and train a custom env by running python3 testRL.py

[parabolang]

Hi everybody
I didn't install sofagym as some build issues I previously in #4305. So I used the DefrostSofaBundle_linux_python3.10_v22.06.01 to try to create RL env and train a model base on observing the sofa scene.
I have created a custom environment in sofa scene by using gymnasium. It is ok to run the gym env and take action by animation function. Things are good until I imported stable_baselines3 library to sofa scene. Each time I ran ./runSofa, errors happened like,
[ERROR] [SofaPython3::SceneLoader] Unable to completely load the scene from file '/home/chi/Desktop/DefrostSofaBundle_linux_python3.10_v22.06.01/blendtest.py'.

Python exception:   
  RuntimeError: function '_has_torch_function' already has a docstring

At:
  /home/chi/miniconda3/envs/soft/lib/python3.10/site-packages/torch/overrides.py(1591): <module>
  <frozen importlib._bootstrap>(241): _call_with_frames_removed
  <frozen importlib._bootstrap_external>(883): exec_module
  <frozen importlib._bootstrap>(703): _load_unlocked
  <frozen importlib._bootstrap>(1006): _find_and_load_unlocked
  <frozen importlib._bootstrap>(1027): _find_and_load
  /home/chi/miniconda3/envs/soft/lib/python3.10/site-packages/torch/_tensor.py(21): <module>
  <frozen importlib._bootstrap>(241): _call_with_frames_removed
  <frozen importlib._bootstrap_external>(883): exec_module
  <frozen importlib._bootstrap>(703): _load_unlocked
  <frozen importlib._bootstrap>(1006): _find_and_load_unlocked
  <frozen importlib._bootstrap>(1027): _find_and_load
  /home/chi/miniconda3/envs/soft/lib/python3.10/site-packages/torch/__init__.py(1119): <module>
  <frozen importlib._bootstrap>(241): _call_with_frames_removed
  <frozen importlib._bootstrap_external>(883): exec_module
  <frozen importlib._bootstrap>(703): _load_unlocked
  <frozen importlib._bootstrap>(1006): _find_and_load_unlocked
  <frozen importlib._bootstrap>(1027): _find_and_load
  /home/chi/miniconda3/envs/soft/lib/python3.10/site-packages/stable_baselines3/a2c/a2c.py(3): <module>
  <frozen importlib._bootstrap>(241): _call_with_frames_removed
  <frozen importlib._bootstrap_external>(883): exec_module
  <frozen importlib._bootstrap>(703): _load_unlocked
  <frozen importlib._bootstrap>(1006): _find_and_load_unlocked
  <frozen importlib._bootstrap>(1027): _find_and_load
  /home/chi/miniconda3/envs/soft/lib/python3.10/site-packages/stable_baselines3/a2c/__init__.py(1): <module>
  <frozen importlib._bootstrap>(241): _call_with_frames_removed
  <frozen importlib._bootstrap_external>(883): exec_module
  <frozen importlib._bootstrap>(703): _load_unlocked
  <frozen importlib._bootstrap>(1006): _find_and_load_unlocked
  <frozen importlib._bootstrap>(1027): _find_and_load
  /home/chi/miniconda3/envs/soft/lib/python3.10/site-packages/stable_baselines3/__init__.py(3): <module>
  <frozen importlib._bootstrap>(241): _call_with_frames_removed
  <frozen importlib._bootstrap_external>(883): exec_module
  <frozen importlib._bootstrap>(703): _load_unlocked
  <frozen importlib._bootstrap>(1006): _find_and_load_unlocked
  <frozen importlib._bootstrap>(1027): _find_and_load
  /home/chi/Desktop/DefrostSofaBundle_linux_python3.10_v22.06.01/blendtest.py(691): createScene

Thanks in advance for your support!!!

[hugtalbot]

Hi @parabolang

Oh ! this is about SofaGym let me poke @samuelmyoussef !

[samuelmyoussef]

Hello,

I'm not exactly sure about this specific error, it could be because of compatibility issues between versions of some of the dependencies with stable baselines3. It's also hard to tell since I don't know how you created your env and how you're importing sb3 in your sofa scene. I recommend trying Hugo's solution to the build issue you're having and using SofaGym afterwards.