From 939cb4d938d2e76447f66ffa8e2abca8c66b5c10 Mon Sep 17 00:00:00 2001
From: Francesco Capuano <francesco.capuano@huggingface.co>
Date: Sat, 27 Sep 2025 12:13:53 +0200
Subject: [PATCH] add: note

---
 src/lerobot/policies/sac/reward_model/modeling_classifier.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/lerobot/policies/sac/reward_model/modeling_classifier.py b/src/lerobot/policies/sac/reward_model/modeling_classifier.py
index dba6a174b..8315796a3 100644
--- a/src/lerobot/policies/sac/reward_model/modeling_classifier.py
+++ b/src/lerobot/policies/sac/reward_model/modeling_classifier.py
@@ -59,7 +59,9 @@ class SpatialLearnedEmbeddings(nn.Module):
         super().__init__()
         self.height = height
         self.width = width
-        self.channel = channel
+        self.channel = (
+            channel  # TODO(fracapuano): this gives issues with non-square images bc is hardcoded to 4
+        )
         self.num_features = num_features
 
         self.kernel = nn.Parameter(torch.empty(channel, height, width, num_features))