From bc96284ca03bd296c8da8b8f065d2e4fba420505 Mon Sep 17 00:00:00 2001 From: Cadene Date: Wed, 24 Apr 2024 16:48:55 +0000 Subject: [PATCH] remove last update --- tests/test_policies.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/test_policies.py b/tests/test_policies.py index 0bc41ec76..2e8242e18 100644 --- 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(