YashashMathur commited on
Commit
3d3f192
·
verified ·
1 Parent(s): 611be05

Clean up duplicate code in graders

Browse files
Files changed (1) hide show
  1. environment/graders.py +0 -28
environment/graders.py CHANGED
@@ -141,34 +141,6 @@ def grade_sql_result(
141
 
142
  return False, 0.01
143
 
144
- # Parse the query result to extract values
145
- # Result format is markdown table: | col1 | col2 |
146
- lines = query_result.strip().split("\n")
147
-
148
- # Skip header and separator lines
149
- data_lines = [l for l in lines if l.strip() and not l.startswith("|---")]
150
-
151
- if len(data_lines) < 2: # Need at least header + 1 data row
152
- return False, 0.0
153
-
154
- # Get the first data row (skip header)
155
- data_row = data_lines[1] if len(data_lines) > 1 else ""
156
-
157
- # Extract values from the row
158
- values = [v.strip() for v in data_row.split("|") if v.strip()]
159
-
160
- if not values:
161
- return False, 0.0
162
-
163
- # For single-value answers, compare the first value
164
- # For multi-column results, try each value
165
- for value in values:
166
- is_correct, score = compare_values(value, ground_truth)
167
- if is_correct:
168
- return True, score
169
-
170
- return False, 0.0
171
-
172
 
173
  def grade_answer(
174
  submitted_answer: str, ground_truth: Any, db_engine: Any = None
 
141
 
142
  return False, 0.01
143
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
144
 
145
  def grade_answer(
146
  submitted_answer: str, ground_truth: Any, db_engine: Any = None