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

Rename class ActorCriticModuleOpt -> ActorCriticOpt

Browse Source
This commit is contained in:
Dominik Jain 2023-11-06 16:50:54 +01:00
parent 5c8d57a2d2
commit 7e6d3d627e
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tianshou/highlevel/agent.py
View File

@ -18,7 +18,7 @@ from tianshou.highlevel.module.core import (
) )
from tianshou.highlevel.module.critic import CriticEnsembleFactory, CriticFactory from tianshou.highlevel.module.critic import CriticEnsembleFactory, CriticFactory
from tianshou.highlevel.module.module_opt import ( from tianshou.highlevel.module.module_opt import (
ActorCriticModuleOpt, ActorCriticOpt,
) )
from tianshou.highlevel.optim import OptimizerFactory from tianshou.highlevel.optim import OptimizerFactory
from tianshou.highlevel.params.policy_params import ( from tianshou.highlevel.params.policy_params import (
@ -297,12 +297,12 @@ class ActorCriticAgentFactory(
envs: Environments, envs: Environments,
device: TDevice, device: TDevice,
lr: float, lr: float,
) -> ActorCriticModuleOpt: ) -> ActorCriticOpt:
actor = self.actor_factory.create_module(envs, device) actor = self.actor_factory.create_module(envs, device)
critic = self.critic_factory.create_module(envs, device, use_action=self.critic_use_action) critic = self.critic_factory.create_module(envs, device, use_action=self.critic_use_action)
actor_critic = ActorCritic(actor, critic) actor_critic = ActorCritic(actor, critic)
optim = self.optim_factory.create_optimizer(actor_critic, lr) optim = self.optim_factory.create_optimizer(actor_critic, lr)
return ActorCriticModuleOpt(actor_critic, optim) return ActorCriticOpt(actor_critic, optim)
@typing.no_type_check @typing.no_type_check
def _create_kwargs(self, envs: Environments, device: TDevice) -> dict[str, Any]: def _create_kwargs(self, envs: Environments, device: TDevice) -> dict[str, Any]:

2
tianshou/highlevel/module/module_opt.py
View File

@ -14,7 +14,7 @@ class ModuleOpt:
@dataclass @dataclass
class ActorCriticModuleOpt: class ActorCriticOpt:
"""Container for an :class:`ActorCritic` instance along with its optimizer.""" """Container for an :class:`ActorCritic` instance along with its optimizer."""
actor_critic_module: ActorCritic actor_critic_module: ActorCritic