[BugFix] Fix forced batch-size in _skip_tensordict

ghstack-source-id: dac84e8b8835e870bce1772d7893c30b6f9af59c Pull Request resolved: #2808
pytorch · Feb 24, 2025 · 3806d1f · 3806d1f
1 parent 5a46379
commit 3806d1f
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/torchrl/envs/common.py b/torchrl/envs/common.py
@@ -1967,7 +1967,6 @@ def select_and_clone(name, x, y):
             select_and_clone,
             next_tensordict,
             device=next_tensordict.device,
-            batch_size=next_tensordict.batch_size,
             default=None,
             filter_empty=True,
             is_leaf=_is_leaf_nontensor,