wax.gym.haiku_env

Gym environment defined from an Haiku module

Classes

HaikuEnv(data_generator, env[, name])

Wrap a pair of init/apply function as a standard python class with state management and Gym env API.