update pinned torchrl version

2024-03-27 12:51:40 -07:00
parent d51feb0e9f
commit 3789fcd5b8
2 changed files with 3 additions and 2 deletions
--- 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
--- 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