(ent_coef)
| 74 | |
| 75 | @pytest.mark.parametrize("ent_coef", ["auto", 0.01, "auto_0.01"]) |
| 76 | def test_sac(ent_coef): |
| 77 | model = SAC( |
| 78 | "MlpPolicy", |
| 79 | "Pendulum-v1", |
| 80 | policy_kwargs=dict(net_arch=[64, 64]), |
| 81 | learning_starts=100, |
| 82 | verbose=1, |
| 83 | buffer_size=250, |
| 84 | ent_coef=ent_coef, |
| 85 | action_noise=NormalActionNoise(np.zeros(1), np.zeros(1)), |
| 86 | ) |
| 87 | model.learn(total_timesteps=200) |
| 88 | |
| 89 | |
| 90 | @pytest.mark.parametrize("n_critics", [1, 3]) |
nothing calls this directly
no test coverage detected
searching dependent graphs…