remove last update

huggingface · Apr 24, 2024 · bc96284 · bc96284
1 parent 3435793
commit bc96284
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/test_policies.py b/tests/test_policies.py
@@ -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(