Skip to content

Commit d604e53

Browse files
author
Vincent Moens
committed
[BugFix] Fix forced batch-size in _skip_tensordict
ghstack-source-id: dac84e8 Pull Request resolved: #2808 (cherry picked from commit 3acf491)
1 parent 9b6ea12 commit d604e53

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

torchrl/envs/common.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1953,7 +1953,6 @@ def select_and_clone(name, x, y):
19531953
select_and_clone,
19541954
next_tensordict,
19551955
device=next_tensordict.device,
1956-
batch_size=next_tensordict.batch_size,
19571956
default=None,
19581957
filter_empty=True,
19591958
is_leaf=_is_leaf_nontensor,

0 commit comments

Comments
 (0)