cannot import name 'flatten_dataclass' from 'trlx.data.ilql_types'
Open
bug
- Dominant language
- Python
- Stars
- 4.8k
- Forks
- 487
- PR merge metrics
- No merged PRs in 30d
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_
Contributor guide
Assessment
This issue has not been assessed yet.