TheRealAIGuy commited on
Commit
060c9f5
·
1 Parent(s): a590d66

Minor fix in inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -209,7 +209,7 @@ def get_model_message(client: OpenAI, step: int, features: list[list[float]]) ->
209
  if len(row) >= 4:
210
  # Matches SYSTEM_PROMPT: 1 if > 0.60, 0 if < 0.30, 1 if in between.
211
  risk_score = row[3]
212
- fallback_decisions.append(0.01 if risk_score < 0.30 else 0.99)
213
  else:
214
  fallback_decisions.append(1)
215
 
 
209
  if len(row) >= 4:
210
  # Matches SYSTEM_PROMPT: 1 if > 0.60, 0 if < 0.30, 1 if in between.
211
  risk_score = row[3]
212
+ fallback_decisions.append(0 if risk_score < 0.30 else 1)
213
  else:
214
  fallback_decisions.append(1)
215