oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Apply suggestions from review

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
src/primaite/game/agent/rewards.py
View File

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

2
src/primaite/notebooks/training_example_ray_multi_agent.ipynb
View File

@@ -83,7 +83,7 @@
"tune.Tuner(\n", "tune.Tuner(\n",
" \"PPO\",\n", " \"PPO\",\n",
" run_config=air.RunConfig(\n", " run_config=air.RunConfig(\n",
" stop={\"timesteps_total\": 511},\n", " stop={\"timesteps_total\": 512},\n",
" ),\n", " ),\n",
" param_space=config\n", " param_space=config\n",
").fit()" ").fit()"

2
src/primaite/session/session.py
View File

@@ -62,7 +62,7 @@ class PrimaiteSession:
def start_session(self) -> None: def start_session(self) -> None:
"""Commence the training/eval session.""" """Commence the training/eval session."""
print("Staring Primaite Session") print("Starting Primaite Session")
self.mode = SessionMode.TRAIN self.mode = SessionMode.TRAIN
n_learn_episodes = self.training_options.n_learn_episodes n_learn_episodes = self.training_options.n_learn_episodes
n_eval_episodes = self.training_options.n_eval_episodes n_eval_episodes = self.training_options.n_eval_episodes