From 53ed84b71259467c064b0abebffb2d970d2de16a Mon Sep 17 00:00:00 2001 From: darkdi Date: Thu, 6 Aug 2026 04:48:44 +0300 Subject: [PATCH] Fix the PPO example docstrings to name the arguments the constructors take --- examples/pytorch/domain_templates/reinforce_learn_ppo.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/examples/pytorch/domain_templates/reinforce_learn_ppo.py b/examples/pytorch/domain_templates/reinforce_learn_ppo.py index 55581c1b68088..1733ed2affb0f 100644 --- a/examples/pytorch/domain_templates/reinforce_learn_ppo.py +++ b/examples/pytorch/domain_templates/reinforce_learn_ppo.py @@ -60,8 +60,7 @@ class ActorCategorical(nn.Module): def __init__(self, actor_net): """ Args: - input_shape: observation shape of the environment - n_actions: number of discrete actions available in the environment + actor_net: network that maps an observation to action logits """ super().__init__() @@ -95,8 +94,8 @@ class ActorContinuous(nn.Module): def __init__(self, actor_net, act_dim): """ Args: - input_shape: observation shape of the environment - n_actions: number of discrete actions available in the environment + actor_net: network that maps an observation to the mean of the action distribution + act_dim: number of continuous action dimensions """ super().__init__() self.actor_net = actor_net