From b1afbccb05b0eeb3c5b4708479a0d8e17fb23308 Mon Sep 17 00:00:00 2001 From: Nicklas Hansen Date: Wed, 27 Mar 2024 12:51:40 -0700 Subject: [PATCH] update pinned torchrl version --- docker/environment.yaml | 4 ++-- tdmpc2/common/buffer.py | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/docker/environment.yaml b/docker/environment.yaml index 2a2e5a5..3081b8d 100644 --- a/docker/environment.yaml +++ b/docker/environment.yaml @@ -39,9 +39,9 @@ dependencies: - protobuf==4.25.2 - pillow==10.2.0 - pyquaternion==0.9.9 - - tensordict-nightly==2024.1.10 + - tensordict-nightly==2024.3.26 - termcolor==2.4.0 - - torchrl-nightly==2024.1.10 + - torchrl-nightly==2024.3.26 - transforms3d==0.4.1 - trimesh==4.0.9 - tqdm==4.66.1 diff --git a/tdmpc2/common/buffer.py b/tdmpc2/common/buffer.py index d601000..decfe94 100644 --- a/tdmpc2/common/buffer.py +++ b/tdmpc2/common/buffer.py @@ -19,6 +19,7 @@ class Buffer(): end_key=None, traj_key='episode', truncated_key=None, + strict_length=True, ) self._batch_size = cfg.batch_size * (cfg.horizon+1) self._num_eps = 0