From ac672f65d112f4b344ab4c29cca78a3360d98067 Mon Sep 17 00:00:00 2001 From: Dominik Jain Date: Mon, 6 Nov 2023 17:17:43 +0100 Subject: [PATCH] Add docstring for ActorFactoryTransientStorageDecorator --- tianshou/highlevel/module/actor.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tianshou/highlevel/module/actor.py b/tianshou/highlevel/module/actor.py index f91fd0d..459e22c 100644 --- a/tianshou/highlevel/module/actor.py +++ b/tianshou/highlevel/module/actor.py @@ -229,6 +229,10 @@ class ActorFactoryDiscreteNet(ActorFactory): class ActorFactoryTransientStorageDecorator(ActorFactory): + """Wraps an actor factory, storing the most recently created actor instance such that it + can be retrieved. + """ + def __init__(self, actor_factory: ActorFactory, actor_future: ActorFuture): self.actor_factory = actor_factory self._actor_future = actor_future