[Test] [TQC] [3/3] Add TQC numerical contracts test

[gtnv]

Summary

Add one focused test file covering pooled target truncation, paper-normalized quantile-Huber loss and gradients, arbitrary batch ranks, nested keys, terminal-NaN masking, priority output, and invalid truncation counts.

cc @vmoens

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/rl/4079

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

⚠️ 15 Awaiting Approval

As of commit 0093352 with merge base d7659c7 ():

AWAITING APPROVAL - The following workflows need approval before CI can run:

• Build Aarch64 Linux Wheels (gh)
• Build Linux Wheels (gh)
• Build M1 Wheels (gh)
• Build Windows Wheels (gh)
• Continuous Benchmark (PR) (gh)
• Generate documentation (gh)
• Habitat Tests on Linux (gh)
• Libs Tests on Linux (gh)
• Lint (gh)
• LLM Tests on Linux (gh)
• Push Binary Nightly (gh)
• SOTA Tests on Linux (gh)
• Tutorials Tests on Linux (gh)
• Unit-tests on Linux (gh)
• Validate Test Partitioning (gh)

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[vmoens]

Reviewed rebased head 009335216aaa16e7dce87bdabdea293f4687dd5b and the current bot/check state.

This PR is not currently self-contained on its declared main base: collecting the only changed file fails with ModuleNotFoundError: No module named torchrl.objectives.tqc, and torchrl/objectives/tqc.py is absent from both this head and freshly fetched origin/main. Since this is labeled 3/3, please either stack/change the base onto the PR that introduces TQC or wait for that dependency to land and rebase; otherwise CI cannot exercise any of these numerical contracts. The new test file should also include the repository-required if __name__ == "__main__": pytest.main(...) block.

Command checked: pytest -q test/objectives/test_tqc.py (collection error above). The numerical coverage itself is well targeted once the production dependency is present.