[Feature] Linearise reward transform#2681

Merged

vmoens merged 17 commits intopytorch:mainfrom louisfaury:lf/agg-rewards-transform

Jan 13, 2025

+453-3

Commits on Jan 3, 2025

Typos
Louis Faury
committed
Add LineariseRewards
Louis Faury
committed
imports
Louis Faury
committed
Tests
Louis Faury
committed
fix shape
Louis Faury
committed
Switch to apply
Louis Faury
committed
Finish fixing tests
Louis Faury
committed
Add example
Louis Faury
committed
Add to doc
Louis Faury
committed

Commits on Jan 8, 2025

Apply suggestions from code review

louisfaury
and
vmoens
authored
Add BoundedContinuous to init
Louis Faury
committed
Add typing for weights
Louis Faury
committed
Correct docstring
Louis Faury
committed
Add reward dtype
Louis Faury
committed

Commits on Jan 9, 2025

Update torchrl/envs/transforms/transforms.py
vmoens
authored

Commits on Jan 10, 2025

Fix test
Louis Faury
committed
Fix test
Louis Faury
committed