This commit is contained in:
Alexander Soare 2024-05-10 07:25:06 +01:00
parent 4ae8d61629
commit 43e1c620b7
1 changed files with 1 additions and 1 deletions

View File

@ -103,7 +103,7 @@ def test_policy(env_name, policy_name, extra_overrides):
dct["observation.image"] = dct["observation.images.top"]
del dct["observation.images.top"]
cfg[keys[0]][keys[1]] = dct
cfg.training.override_dataset_stats = None
cfg.override_dataset_stats = None
# Check that we can make the policy object.
dataset = make_dataset(cfg)