Tweaks to make training data handling functions work with new code structure.
Testing
How did you make sure this worked? How can a reviewer verify this?
To-do list
- [ ] add other TODO items here if necessary! questions that need to answered, decisions that need to be made, tests that need to be run, etc.
- [ ] Update relevant documentation - like comments, docstrings, README, release notes, etc.
- [ ] Review the PR yourself and call out any questions or issues you have
Overview
Closes #78 .
What problem does this address?
What did you change in this PR?
Testing
How did you make sure this worked? How can a reviewer verify this?
To-do list