Toy example error

[KehanLong]

Hi,

Thank you for sharing the nice project. I tried to run the toy example with the given instructions, but got an assertion error complaining about R.shape[1] and x_dim does not match. It seems that R.shape[1] = 2 while x_dim = 1. Could you check and let me know where I may make a mistake?

Thank you in advance.

Attached is the output for your reference:

[hongkai-dai]

Thanks a lot for your interest in the project and reporting the error. That was a bug in the code. I just fixed it in the upstream. Let me know if you run into other issues.