Remove batch length restrictions in select_action (#123)

This commit is contained in:
Simon Alibert
2024-05-04 15:33:55 +02:00
committed by GitHub
parent bccee745c3
commit c015252e20
3 changed files with 3 additions and 2 deletions

View File

@@ -71,6 +71,9 @@ class ACTPolicy(nn.Module, PyTorchModelHubMixin):
environment. It works by managing the actions in a queue and only calling `select_actions` when the
queue is empty.
"""
assert "observation.images.top" in batch
assert "observation.state" in batch
self.eval()
batch = self.normalize_inputs(batch)