From bcc98b4b5f1d45b33b2de64b32dc334ddb1b96d3 Mon Sep 17 00:00:00 2001
From: Khalil Meftah <khalil.meftah@huggingface.co>
Date: Fri, 17 Apr 2026 15:10:01 +0200
Subject: [PATCH] refactor(rewards): expand __init__ facade and fix SARMConfig
 __post_init__ crash

---
 src/lerobot/rewards/__init__.py                | 15 +++++++++++++++
 src/lerobot/rewards/sarm/configuration_sarm.py |  2 --
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/src/lerobot/rewards/__init__.py b/src/lerobot/rewards/__init__.py
index 54096879d..203fe2ee1 100644
--- a/src/lerobot/rewards/__init__.py
+++ b/src/lerobot/rewards/__init__.py
@@ -13,9 +13,24 @@
 # limitations under the License.
 
 from .classifier.configuration_classifier import RewardClassifierConfig as RewardClassifierConfig
+from .factory import (
+    get_reward_model_class as get_reward_model_class,
+    make_reward_model as make_reward_model,
+    make_reward_model_config as make_reward_model_config,
+    make_reward_pre_post_processors as make_reward_pre_post_processors,
+)
+from .pretrained import PreTrainedRewardModel as PreTrainedRewardModel
 from .sarm.configuration_sarm import SARMConfig as SARMConfig
 
 __all__ = [
+    # Configuration classes
     "RewardClassifierConfig",
     "SARMConfig",
+    # Base class
+    "PreTrainedRewardModel",
+    # Factory functions
+    "get_reward_model_class",
+    "make_reward_model",
+    "make_reward_model_config",
+    "make_reward_pre_post_processors",
 ]
diff --git a/src/lerobot/rewards/sarm/configuration_sarm.py b/src/lerobot/rewards/sarm/configuration_sarm.py
index fcb2043a3..b06979c8e 100644
--- a/src/lerobot/rewards/sarm/configuration_sarm.py
+++ b/src/lerobot/rewards/sarm/configuration_sarm.py
@@ -108,8 +108,6 @@ class SARMConfig(RewardModelConfig):
     )
 
     def __post_init__(self):
-        super().__post_init__()
-
         if self.annotation_mode not in ["single_stage", "dense_only", "dual"]:
             raise ValueError(
                 f"annotation_mode must be 'single_stage', 'dense_only', or 'dual', got {self.annotation_mode}"