running test.py for VAIL

[ranahanocka]

Hello,

I am running the VAIL example as explained in the README.

main.py works as expected.

When running test (e.g., as written: python test.py --load_model ckpt_4000_vail.pth.tar), there is an import error.

Specifically:

lets-do-irl/mujoco/vail/test.py

Line 8 in ad04bf1

from utils.running_state import ZFilter

there is no file running_state. In the main.py the running state was defined from Zfilter.

[csufangyu]

Hello,

I am running the VAIL example as explained in the README.

main.py works as expected.

When running test (e.g., as written: python test.py --load_model ckpt_4000_vail.pth.tar), there is an import error.

Specifically:

lets-do-irl/mujoco/vail/test.py

Line 8 in ad04bf1

from utils.running_state import ZFilter

there is no file running_state. In the main.py the running state was defined from Zfilter.

maybe you should change it.the Zfliter is in utils.zfilter.you can change it as :
from utils.zfilter import ZFilter

[ranahanocka]

Thanks @csufangyu. I previously modified the code to make it work, but thought the authors might want to fix the repo