Skip to content

Commit 4e8f08c

Browse files
authored
Refactor TFPolicy and Policy (#4254)
* Refactor TFPolicy and Policy
1 parent 14fad19 commit 4e8f08c

File tree

12 files changed

+409
-375
lines changed

12 files changed

+409
-375
lines changed

ml-agents/mlagents/trainers/components/reward_signals/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def __init__(self, policy: TFPolicy, settings: RewardSignalSettings):
2323
"""
2424
Initializes a reward signal. At minimum, you must pass in the policy it is being applied to,
2525
the reward strength, and the gamma (discount factor.)
26-
:param policy: The Policy object (e.g. NNPolicy) that this Reward Signal will apply to.
26+
:param policy: The Policy object (e.g. TFPolicy) that this Reward Signal will apply to.
2727
:param settings: Settings parameters for this Reward Signal, including gamma and strength.
2828
:return: A RewardSignal object.
2929
"""

ml-agents/mlagents/trainers/policy/nn_policy.py

Lines changed: 0 additions & 275 deletions
This file was deleted.

0 commit comments

Comments
 (0)