#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 a35c363345
commit bb9bfc50a5
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)