#1386: remove setting of global seed + running pre-commit checks
This commit is contained in:
@@ -248,7 +248,6 @@ class AgentSessionABC(ABC):
|
||||
|
||||
agent.session_path = path
|
||||
|
||||
|
||||
return agent
|
||||
|
||||
else:
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user