Apply suggestions from review

This commit is contained in:
Marek Wolan
2023-12-04 09:21:47 +00:00
parent 105763971d
commit ba3d37316b
3 changed files with 3 additions and 3 deletions

View File

@@ -238,7 +238,7 @@ class RewardFunction:
"""Initialise the reward function object."""
self.reward_components: List[Tuple[AbstractReward, float]] = []
"attribute reward_components keeps track of reward components and the weights assigned to each."
self.current_reward: float
self.current_reward: float = 0.0
self.total_reward: float = 0.0
def regsiter_component(self, component: AbstractReward, weight: float = 1.0) -> None: