What's Changed
* Keyframe get / set state does not return the right keys by smassonnet in https://github.com/LSIR/mlmodule/pull/80
* Improve dataset modularity by smassonnet in https://github.com/LSIR/mlmodule/pull/82
* Remove logic to filter out features with NaN by smassonnet in https://github.com/LSIR/mlmodule/pull/83
**Full Changelog**: https://github.com/LSIR/mlmodule/compare/v0.6.0a1...v0.6.0a2