remove last update

This commit is contained in:
Cadene 2024-04-24 16:48:55 +00:00
parent 343579368b
commit bc96284ca0
1 changed files with 0 additions and 1 deletions

View File

@ -100,7 +100,6 @@ def test_policy(env_name, policy_name, extra_overrides):
# TODO(rcadene, alexander-soar): how to remove need for dataset_stats?
new_policy = make_policy(cfg, dataset_stats=dataset.stats)
new_policy.load_state_dict(policy.state_dict())
new_policy.update(batch, step=0)
@pytest.mark.parametrize(