[RLlib] Fix typo in docstring of PGTorchPolicy (#23818)

ray-project · Apr 11, 2022 · c619104 · c619104
1 parent a3d4fc7
commit c619104
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rllib/agents/pg/pg_torch_policy.py b/rllib/agents/pg/pg_torch_policy.py
@@ -72,7 +72,7 @@ def pg_loss_stats(policy: Policy, train_batch: SampleBatch) -> Dict[str, TensorT
     }
 
 
-# Build a child class of `TFPolicy`, given the extra options:
+# Build a child class of `TorchPolicy`, given the extra options:
 # - trajectory post-processing function (to calculate advantages)
 # - PG loss function
 PGTorchPolicy = build_policy_class(