#1386: remove setting of global seed + running pre-commit checks

This commit is contained in:
Czar Echavez
2023-07-06 12:10:26 +01:00
parent 3438ce7e09
commit 99f1f7cfc1
4 changed files with 9 additions and 27 deletions

View File

@@ -248,7 +248,6 @@ class AgentSessionABC(ABC):
agent.session_path = path
return agent
else:

View File

@@ -59,7 +59,7 @@ class SB3Agent(AgentSessionABC):
verbose=self.sb3_output_verbose_level,
n_steps=self._training_config.num_steps,
tensorboard_log=str(self._tensorboard_log_path),
seed=self._training_config.seed
seed=self._training_config.seed,
)
def _save_checkpoint(self):
@@ -121,10 +121,7 @@ class SB3Agent(AgentSessionABC):
obs = self._env.reset()
for step in range(time_steps):
action, _states = self._agent.predict(
obs,
deterministic=self._training_config.deterministic
)
action, _states = self._agent.predict(obs, deterministic=self._training_config.deterministic)
if isinstance(action, np.ndarray):
action = np.int64(action)
obs, rewards, done, info = self._env.step(action)

View File

@@ -2,7 +2,6 @@
"""Main environment module containing the PRIMmary AI Training Evironment (Primaite) class."""
import copy
import logging
import random
import uuid as uuid
from pathlib import Path
from random import choice, randint, sample, uniform
@@ -241,11 +240,6 @@ class Primaite(Env):
self.episode_av_reward_writer = SessionOutputWriter(self, transaction_writer=False, learning_session=True)
self.transaction_writer = SessionOutputWriter(self, transaction_writer=True, learning_session=True)
# set the seed globally if there is one
if self.training_config.seed:
random.seed(self.training_config.seed)
np.random.seed(self.training_config.seed)
@property
def actual_episode_count(self) -> int:
"""Shifts the episode_count by -1 for RLlib."""