Read more parameters from `env.yaml`

Currently, if we change the policy API (like observation shape, frequency, etc.) we have to update the parameters manually in the inference code. We should read these values from the config file, so the user can easily replace the policy.