anugrah55 commited on
Commit
17d2d79
·
verified ·
1 Parent(s): 00601c1

Fix inference json fallback logic

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -76,7 +76,7 @@ def get_model_action(client: OpenAI, obs_dict: dict) -> dict:
76
  return json.loads(text.strip())
77
  except Exception as exc:
78
  print(f"[DEBUG] Model request failed: {exc}", flush=True)
79
- return {"action_type": "submit"}
80
 
81
  async def run_task(task_name: str, client: OpenAI, env_client) -> None:
82
  log_start(task=task_name, env=BENCHMARK, model=MODEL_NAME)
 
76
  return json.loads(text.strip())
77
  except Exception as exc:
78
  print(f"[DEBUG] Model request failed: {exc}", flush=True)
79
+ return {"action_type": "fill_na", "column_name": "invalid", "value": "invalid"}
80
 
81
  async def run_task(task_name: str, client: OpenAI, env_client) -> None:
82
  log_start(task=task_name, env=BENCHMARK, model=MODEL_NAME)