fix(smolvla): update record.py, fix populate_queues and remove unused dependencies (#1208)

This commit is contained in:
Steven Palma
2025-06-06 09:17:02 +02:00
committed by GitHub
parent e23b41e79a
commit 09343acce7
2 changed files with 9 additions and 3 deletions

View File

@@ -14,15 +14,21 @@
# See the License for the specific language governing permissions and
# limitations under the License.
from collections import deque
import torch
from torch import nn
def populate_queues(queues, batch):
def populate_queues(
queues: dict[str, deque], batch: dict[str, torch.Tensor], exclude_keys: list[str] | None = None
):
if exclude_keys is None:
exclude_keys = []
for key in batch:
# Ignore keys not in the queues already (leaving the responsibility to the caller to make sure the
# queues have the keys they want).
if key not in queues:
if key not in queues or key in exclude_keys:
continue
if len(queues[key]) != queues[key].maxlen:
# initialize by copying the first observation several times until the queue is full