Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
michalzajac-ml committed Sep 7, 2023
1 parent 170db55 commit b12e9aa
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions tests/algorithms/test_sqil.py
Original file line number Diff line number Diff line change
Expand Up @@ -182,7 +182,6 @@ def _test_sqil_performance(
pytestconfig: pytest.Config,
venv: vec_env.VecEnv,
env_name: str,
total_timesteps: int,
rl_algo_class: Type[off_policy_algorithm.OffPolicyAlgorithm],
rl_kwargs: Optional[Dict[str, Any]] = None,
):
Expand All @@ -204,7 +203,7 @@ def _test_sqil_performance(
return_episode_rewards=True,
)

model.train(total_timesteps=total_timesteps)
model.train(total_timesteps=10_000)

venv.seed(SEED)
rewards_after, _ = evaluate_policy(
Expand All @@ -230,7 +229,6 @@ def test_sqil_performance_discrete(
pytestconfig,
cartpole_venv,
"seals/CartPole-v0",
total_timesteps=10_000,
rl_algo_class=dqn.DQN,
rl_kwargs=dict(
learning_starts=500,
Expand All @@ -254,7 +252,6 @@ def test_sqil_performance_continuous(
pytestconfig,
pendulum_single_venv,
"Pendulum-v1",
total_timesteps=20_000,
rl_algo_class=rl_algo_class,
rl_kwargs=dict(
learning_rate=0.002,
Expand Down

0 comments on commit b12e9aa

Please sign in to comment.