gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-29 00:17:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix bug in model evaluator final feedback ignoring shape feedback (#501)

Browse Source
This commit is contained in:
Haoran Pan
2024-12-03 13:55:28 +08:00
committed by GitHub
parent d4e7f9277e
commit 95d81fa71b
3 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/model_coder/eva_utils.py
+2
View File
@@ -123,6 +123,7 @@ class ModelFinalEvaluator(Evaluator):
implementation: Workspace,
gt_implementation: Workspace,
model_execution_feedback: str,
model_shape_feedback: str,
model_value_feedback: str,
model_code_feedback: str,
):
@@ -154,6 +155,7 @@ class ModelFinalEvaluator(Evaluator):
.render(
model_information=target_task.get_task_information(),
model_execution_feedback=execution_feedback_to_render,
model_shape_feedback=model_shape_feedback,
model_code_feedback=model_code_feedback,
model_value_feedback=model_value_feedback,
)