Simplifying the IO logic and removing a helper.
Removed unused import.
Merge pull request #855 from rhyolight/one-hot-gym-prediction

Simplifying the IO logic and removing a helper.