Skip to content

[BugFix] Preserve lazy cat for mixed TensorDict inputs - #1732

Merged
vmoens merged 1 commit into
pytorch:mainfrom
vmoens:codex/mixed-lazy-plain-cat
Jul 6, 2026
Merged

[BugFix] Preserve lazy cat for mixed TensorDict inputs#1732
vmoens merged 1 commit into
pytorch:mainfrom
vmoens:codex/mixed-lazy-plain-cat

Conversation

@vmoens

@vmoens vmoens commented Jul 6, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • route mixed lazy/plain TensorDict cat through lazy cat normalization
  • convert dense inputs to compatible lazy stacks instead of densifying lazy stacks
  • add coverage for stack-dim, non-stack-dim, non-densifiable, zero-length, and dense out cases

Tests

  • uv run --extra tests python -m pytest test/tensordict/test_lazy.py -q

@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Jul 6, 2026
@github-actions github-actions Bot added bug Something isn't working Test and removed bug Something isn't working labels Jul 6, 2026
@vmoens
vmoens force-pushed the codex/mixed-lazy-plain-cat branch from 8ef3f90 to 8894d54 Compare July 6, 2026 10:23
@github-actions github-actions Bot added the bug Something isn't working label Jul 6, 2026
@vmoens
vmoens merged commit 045eeed into pytorch:main Jul 6, 2026
63 of 66 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. Test

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant