cannot import name 'flatten_dataclass' from 'trlx.data.ilql_types' #599
Description
🐛 Describe the bug
I am trying to run examples/nemo_ppo_sentiments.py but I get this error
cannot import name 'flatten_dataclass' from 'trlx.data.ilql_types'
It seems that there is no flatten_dataclass or unflatten_dataclass in ilql_types.
Which trlX version are you using?
No response
Additional system and package information
No response