ImportError: cannot import name 'CollectStats' from 'tianshou.data'

[hmishfaq]

I am using the latest version of tianshou and having the following error:

/tianshou/test/discrete$ python test_sac.py 
Traceback (most recent call last):
  File "/home/mila/i/ishfaqha/tianshou/test/discrete/test_sac.py", line 9, in <module>
    from tianshou.data import Collector, CollectStats, VectorReplayBuffer
ImportError: cannot import name 'CollectStats' from 'tianshou.data' (/home/mila/i/ishfaqha/envs/redq/lib/python3.10/site-packages/tianshou/data/__init__.py)

How can I fix it?

[Trinkle23897]

could you double check you are on the new version?