fix test
This commit is contained in:
parent
4ae8d61629
commit
43e1c620b7
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue