From 26d1668f7ddd815679ccb0b71e0054dca372a147 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Mon, 28 Oct 2024 21:39:39 -0700
Subject: [PATCH] Revert "remove Field for return_type"

This reverts commit ffb3965ade73b5a0219ecaeb2ef873d5c88b8066.
---
 llama_stack/apis/scoring_functions/scoring_functions.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/llama_stack/apis/scoring_functions/scoring_functions.py b/llama_stack/apis/scoring_functions/scoring_functions.py
index 6b90aea6d..2e5bf0aef 100644
--- a/llama_stack/apis/scoring_functions/scoring_functions.py
+++ b/llama_stack/apis/scoring_functions/scoring_functions.py
@@ -44,7 +44,9 @@ class ScoringFnDef(BaseModel):
         description="List of parameters for the deterministic function",
         default_factory=list,
     )
-    return_type: ParamType
+    return_type: ParamType = Field(
+        description="The return type of the deterministic function",
+    )
     context: Optional[LLMAsJudgeContext] = None
     # We can optionally add information here to support packaging of code, etc.