hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add docstring for ActorFactoryTransientStorageDecorator

Browse Source
This commit is contained in:
Dominik Jain 2023-11-06 17:17:43 +01:00
parent 7e6d3d627e
commit ac672f65d1
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tianshou/highlevel/module/actor.py
View File

@ -229,6 +229,10 @@ class ActorFactoryDiscreteNet(ActorFactory):
class ActorFactoryTransientStorageDecorator(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): def __init__(self, actor_factory: ActorFactory, actor_future: ActorFuture):
self.actor_factory = actor_factory self.actor_factory = actor_factory
self._actor_future = actor_future self._actor_future = actor_future