Change all signatures for policy.process_fn and policy.post_process_fn from BatchProtocol to RolloutBatchProtocol

thu-ml / tianshou

An elegant PyTorch deep reinforcement learning library.

https://tianshou.org

MIT License

7.82k stars 1.12k forks source link

Change all signatures for policy.process_fn and policy.post_process_fn from BatchProtocol to RolloutBatchProtocol #919

Open MischaPanch opened 1 year ago

MischaPanch commented 1 year ago

This is purely a typing issue

There might also a possibility/necessity to use stricter typing, dependent on what fields the process_fn and post_process_fn require from the Batch objects in the various implementations

arnaujc91 commented 9 months ago

Will take a look at this one