From b22c35e34b29666391a70e6546248b8b60c1cbc6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gabriel=20Mart=C3=ADn=20Bl=C3=A1zquez?=
<gmartinbdev@gmail.com>
Date: Thu, 15 Aug 2024 18:59:43 +0200
Subject: [PATCH] Update `RewardModelScore.inputs`
---
src/distilabel/steps/reward_model.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/src/distilabel/steps/reward_model.py b/src/distilabel/steps/reward_model.py
index b6d68d0f47..9a88d8d659 100644
--- a/src/distilabel/steps/reward_model.py
+++ b/src/distilabel/steps/reward_model.py
@@ -181,7 +181,11 @@ def load(self) -> None:
@property
def inputs(self) -> "StepColumns":
"""Either `response` and `instruction`, or a `conversation` columns."""
- return []
+ return {
+ "response": False,
+ "instruction": False,
+ "conversation": False,
+ }
@property
def outputs(self) -> "StepColumns":