Skip to content

Commit

Permalink
correct skill type hint
Browse files Browse the repository at this point in the history
  • Loading branch information
kywch committed Sep 4, 2023
1 parent 1b864fe commit 6a92799
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions nmmo/task/base_predicates.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,8 @@ def DistanceTraveled(gs: GameState, subject: Group, dist: int):
dists = utils.linf(list(zip(r,c)),[gs.spawn_pos[id_] for id_ in subject.entity.id])
return norm(dists.sum() / dist)

def AttainSkill(gs: GameState, subject: Group, skill: Skill, level: int, num_agent: int):
def AttainSkill(gs: GameState, subject: Group,
skill: type[Skill], level: int, num_agent: int):
"""True if the number of agents having skill level GE level
is greather than or equal to num_agent
"""
Expand All @@ -88,7 +89,8 @@ def AttainSkill(gs: GameState, subject: Group, skill: Skill, level: int, num_age
skill_level = getattr(subject,skill.__name__.lower() + '_level') - 1 # base level is 1
return norm(sum(skill_level) / (num_agent * (level-1)))

def GainExperience(gs: GameState, subject: Group, skill: Skill, experience: int, num_agent: int):
def GainExperience(gs: GameState, subject: Group,
skill: type[Skill], experience: int, num_agent: int):
"""True if the experience gained for the skill is greater than or equal to experience."""
skill_exp = getattr(subject,skill.__name__.lower() + '_exp')
return norm(sum(skill_exp) / (experience*num_agent))
Expand Down

0 comments on commit 6a92799

Please sign in to comment.