This repository has been archived by the owner on Sep 1, 2024. It is now read-only.
v0.1.5
- Fixes important bug in v0.1.4 that was causing PETS to break.
Model.reset()
andModel.sample()
signature has changed. They no longer receive
TransitionBatch
objects, and they both return a dictionary of strings to tensors
representing a model state that should be passed tosample()
to simulate
transitions. This dictionary can contain things like previous actions, predicted
observation, latent states, beliefs, and any other such quantity that the model
need to maintain to simulate trajectories when usingModelEnv
.Ensemble
class and sub-classes are assumed to operate on 1-D models.