Skip to content

viewer.launch -- Add support for reset-callback of policy #323

@simon-bachhuber

Description

@simon-bachhuber

Hello,

right now the viewer.launch function expects its policy parameter to be a function mapping observation to action.
When pressing the Return-key the environment resets but the policy might also have some state that should reset.
How can this functionality be added?

Right now the only workaround i could think of would be to explicitly padd time to the observation which seems a bit weird since it requires a modification of the environment itself.

Thank you!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions