Can't run example.py following to README

[koseoyoung]

Hi, I'm trying to run example.py according to the README.

(s4-plain) [sko1@v015 state-spaces]$ python example.py
/jet/home/sko1/.conda/envs/s4-plain/lib/python3.9/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory
  warn(f"Failed to load image Python extension: {e}")
Traceback (most recent call last):
  File "/ocean/projects/cis230033p/sko1/state-spaces/example.py", line 35, in <module>
    from models.s4.s4 import S4
ImportError: cannot import name 'S4' from 'models.s4.s4' (/ocean/projects/cis230033p/sko1/state-spaces/models/s4/s4.py)

[albertfgu]

Thanks for the catch, it's importing the wrong symbol after the latest v4 changes. Actually this module is not used at all in example.py, so you can just comment out this line. I'm pushing a fix soon.

[albertfgu]

Pushed