From 32e7ae2a5bc97043779cc6cfc5c59e3561f15ecd Mon Sep 17 00:00:00 2001
From: Mehdi Lotifan <79265020+MehdiLotfian@users.noreply.github.com>
Date: Tue, 17 Mar 2026 18:54:54 +0800
Subject: [PATCH] Handle None input in sanitize function

---
 bigcodebench/sanitize.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/bigcodebench/sanitize.py b/bigcodebench/sanitize.py
index 2f8ba3b..5215745 100644
--- a/bigcodebench/sanitize.py
+++ b/bigcodebench/sanitize.py
@@ -180,6 +180,8 @@ def extract_target_code_or_empty(code: str, entrypoint: Optional[str] = None) ->
 
 
 def sanitize(code: str, entrypoint: Optional[str] = None) -> str:
+    if code is None:
+        return ""
     sanitized_code = extract_target_code_or_empty(code, entrypoint).strip()
     if not sanitized_code:
         return code_extract(code)