Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion tasks/agentic-search-vectorless/tests/run_submission.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,8 @@

def submission_env() -> dict[str, str]:
env = {"HOME": "/tmp/agentic-search-vectorless-submission-home", "TMPDIR": "/tmp",
"PATH": "/opt/conda/bin:/usr/local/bin:/usr/bin:/bin", "LANG": "C.UTF-8",
"JAVA_HOME": "/opt/java/openjdk",
"PATH": "/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin", "LANG": "C.UTF-8",
"TOKENIZERS_PARALLELISM": "false", "PYTHONUNBUFFERED": "1"}
for key in AGENT_API_NAMES | {"HTTP_PROXY", "HTTPS_PROXY", "NO_PROXY",
"http_proxy", "https_proxy", "no_proxy"}:
Expand Down
1 change: 1 addition & 0 deletions tasks/agentic-search/tests/run_submission.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,7 @@ def submission_env() -> dict[str, str]:
# Judge/coding-agent credentials and endpoint settings never cross this boundary.
names = AGENT_API_NAMES | {"PATH", "LANG", "LC_ALL", "PYTHONUNBUFFERED", "REACT_MODEL"}
env = {name: os.environ[name] for name in names if name in os.environ}
env["JAVA_HOME"] = "/opt/java/openjdk"
env["HOME"] = "/tmp/agentic-search-submission-home"
env["TMPDIR"] = "/tmp"
return env
Expand Down
3 changes: 2 additions & 1 deletion tasks/biomedical-graph-retrieval/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -126,7 +126,8 @@ def submission_command(command: list[str]) -> list[str]:
f"HOME={INDEX_DIR}",
"USER=submission",
"LOGNAME=submission",
"PATH=/opt/conda/bin:/usr/local/bin:/usr/bin:/bin",
"JAVA_HOME=/opt/java/openjdk",
"PATH=/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin",
"LANG=C.UTF-8",
"PYTHONUNBUFFERED=1",
"PYTHONDONTWRITEBYTECODE=1",
Expand Down
3 changes: 2 additions & 1 deletion tasks/budgeted-bimetric-retrieval/tests/runtime.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,7 +118,8 @@ def __init__(self, command, work, log, cpus, fixed_bytes=0):
self.log = log.open('wb', buffering=0)
self.logged = 0
scratch = str(Path(work)/'scratch') if (Path(work)/'scratch').exists() else work
env = {'PATH': '/opt/conda/bin:/usr/local/bin:/usr/bin:/bin', 'HOME': scratch,
env = {'JAVA_HOME': '/opt/java/openjdk',
'PATH': '/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin', 'HOME': scratch,
'TMPDIR': scratch, 'XDG_CACHE_HOME': scratch, 'PYTHONDONTWRITEBYTECODE': '1',
'PYTHONUNBUFFERED': '1', 'LANG': 'C.UTF-8', 'OMP_NUM_THREADS': str(len(cpus)),
'OPENBLAS_NUM_THREADS': str(len(cpus)), 'MKL_NUM_THREADS': str(len(cpus)),
Expand Down
3 changes: 2 additions & 1 deletion tasks/budgeted-candidate-reranking/tests/runtime/evaluate.py
Original file line number Diff line number Diff line change
Expand Up @@ -245,7 +245,8 @@ def run_query(query, submission, backend, output, resources):
excluded = 0
stderr = (output / (query['query_id']+'.stderr')).open('wb')
cpus = sorted(os.sched_getaffinity(0))[:16]
env = {'PATH': '/opt/conda/bin:/usr/local/bin:/usr/bin:/bin', 'HOME': '/work',
env = {'JAVA_HOME': '/opt/java/openjdk',
'PATH': '/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin', 'HOME': '/work',
'TMPDIR': '/work', 'XDG_CACHE_HOME': '/work', 'LANG': 'C.UTF-8',
'PYTHONDONTWRITEBYTECODE': '1', 'PYTHONUNBUFFERED': '1',
'OMP_NUM_THREADS': str(len(cpus)), 'OPENBLAS_NUM_THREADS': str(len(cpus))}
Expand Down
3 changes: 2 additions & 1 deletion tasks/chunqiu-temporal-retrieval/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,8 @@ def read_jsonl(path):


def child_command(command):
environment = ["PATH=/opt/conda/bin:/usr/local/bin:/usr/bin:/bin", "USER=submission",
environment = ["JAVA_HOME=/opt/java/openjdk",
"PATH=/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin", "USER=submission",
"LOGNAME=submission", f"HOME={INDEX}", "LANG=C.UTF-8", "PYTHONUNBUFFERED=1",
"PYTHONDONTWRITEBYTECODE=1", "TOKENIZERS_PARALLELISM=false",
"OMP_NUM_THREADS=8", "OPENBLAS_NUM_THREADS=8", "HF_HUB_OFFLINE=1",
Expand Down
3 changes: 2 additions & 1 deletion tasks/code-context-retrieval/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,8 @@ def prepare():

def child_env():
env = {'HOME': str(INDEX), 'USER': 'submission', 'LOGNAME': 'submission',
'PATH': '/opt/conda/bin:/usr/local/bin:/usr/bin:/bin', 'LANG': 'C.UTF-8',
'JAVA_HOME': '/opt/java/openjdk',
'PATH': '/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin', 'LANG': 'C.UTF-8',
'TMPDIR': str(INDEX / 'tmp'), 'XDG_CACHE_HOME': str(INDEX / 'cache'),
'PYTHONUNBUFFERED': '1', 'PYTHONDONTWRITEBYTECODE': '1',
'OMP_NUM_THREADS': '16', 'OPENBLAS_NUM_THREADS': '16',
Expand Down
3 changes: 2 additions & 1 deletion tasks/composed-image-search/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,8 @@ def child_env():
"HOME": "/work/index",
"USER": "submission",
"LOGNAME": "submission",
"PATH": "/opt/conda/bin:/usr/local/bin:/usr/bin:/bin",
"JAVA_HOME": "/opt/java/openjdk",
"PATH": "/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin",
"LANG": "C.UTF-8",
"PYTHONUNBUFFERED": "1",
"PYTHONDONTWRITEBYTECODE": "1",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -187,7 +187,7 @@ def __init__(self, work, output, resources, library_dir):
"PATH", "LD_LIBRARY_PATH", "CUDA_VISIBLE_DEVICES", "NVIDIA_VISIBLE_DEVICES",
"NVIDIA_DRIVER_CAPABILITIES") if name in os.environ}
scratch = work / "scratch" if (work / "scratch").exists() else work
env.update(HOME=str(scratch), TMPDIR=str(scratch), XDG_CACHE_HOME=str(scratch),
env.update(JAVA_HOME="/opt/java/openjdk", HOME=str(scratch), TMPDIR=str(scratch), XDG_CACHE_HOME=str(scratch),
MOLECULE_WORK_DIR=str(work), MOLECULE_LIBRARY_DIR=str(library_dir),
PYTHONUNBUFFERED="1", PYTHONDONTWRITEBYTECODE="1",
HF_HUB_OFFLINE="1", TRANSFORMERS_OFFLINE="1", OMP_NUM_THREADS="16",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -187,7 +187,7 @@ def __init__(self, work, output, resources, library_dir):
"PATH", "LD_LIBRARY_PATH", "CUDA_VISIBLE_DEVICES", "NVIDIA_VISIBLE_DEVICES",
"NVIDIA_DRIVER_CAPABILITIES") if name in os.environ}
scratch = work / "scratch" if (work / "scratch").exists() else work
env.update(HOME=str(scratch), TMPDIR=str(scratch), XDG_CACHE_HOME=str(scratch),
env.update(JAVA_HOME="/opt/java/openjdk", HOME=str(scratch), TMPDIR=str(scratch), XDG_CACHE_HOME=str(scratch),
MOLECULE_WORK_DIR=str(work), MOLECULE_LIBRARY_DIR=str(library_dir),
PYTHONUNBUFFERED="1", PYTHONDONTWRITEBYTECODE="1",
HF_HUB_OFFLINE="1", TRANSFORMERS_OFFLINE="1", OMP_NUM_THREADS="16",
Expand Down
3 changes: 2 additions & 1 deletion tasks/decision-model-training/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,8 @@ def stop_children():


def child_command(command, runtime):
environment = {"PATH": "/opt/conda/bin:/usr/local/bin:/usr/bin:/bin",
environment = {"JAVA_HOME": "/opt/java/openjdk",
"PATH": "/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin",
"USER": "submission", "LOGNAME": "submission", "HOME": str(runtime),
"LANG": "C.UTF-8", "PYTHONUNBUFFERED": "1", "PYTHONDONTWRITEBYTECODE": "1",
"OMP_NUM_THREADS": "16", "MKL_NUM_THREADS": "16", "OPENBLAS_NUM_THREADS": "1",
Expand Down
3 changes: 2 additions & 1 deletion tasks/document-page-search/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,8 @@ def child_env():
"HOME": "/work/index",
"USER": "submission",
"LOGNAME": "submission",
"PATH": "/opt/conda/bin:/usr/local/bin:/usr/bin:/bin",
"JAVA_HOME": "/opt/java/openjdk",
"PATH": "/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin",
"LANG": "C.UTF-8",
"PYTHONUNBUFFERED": "1",
"PYTHONDONTWRITEBYTECODE": "1",
Expand Down
3 changes: 2 additions & 1 deletion tasks/evolving-personal-memory/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,8 @@ def child_env(home, answer=False):
"HOME": str(home),
"USER": "answer" if answer else "submission",
"LOGNAME": "answer" if answer else "submission",
"PATH": "/opt/conda/bin:/usr/local/bin:/usr/bin:/bin",
"JAVA_HOME": "/opt/java/openjdk",
"PATH": "/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin",
"LANG": "C.UTF-8",
"PYTHONUNBUFFERED": "1",
"PYTHONDONTWRITEBYTECODE": "1",
Expand Down
3 changes: 2 additions & 1 deletion tasks/filtered-ann-retrieval/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -121,7 +121,8 @@ def __init__(self, command, app, work, log, run_as, threads, *, service=False):
self.run_as = run_as
self.log = log.open("wb")
environment = {
"PATH": f"{Path(sys.executable).parent}:/usr/local/bin:/usr/bin:/bin",
"JAVA_HOME": "/opt/java/openjdk",
"PATH": f"{Path(sys.executable).parent}:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin",
"HOME": str(work), "TMPDIR": str(work / "tmp"),
"LANG": "C.UTF-8", "LC_ALL": "C.UTF-8",
"PYTHONUNBUFFERED": "1", "PYTHONDONTWRITEBYTECODE": "1",
Expand Down
2 changes: 1 addition & 1 deletion tasks/filtered-ann-retrieval/tests/offline_exec.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@ def allow(path, access):
allow(index, writable)
# Interpreter, shared libraries and read-only system configuration.
# Do not allow /task, /tmp, /home, /dev/shm or a common ancestor of index.
for path in ("/usr", "/bin", "/sbin", "/lib", "/lib64", "/opt/conda",
for path in ("/usr", "/bin", "/sbin", "/lib", "/lib64", "/opt/conda", "/opt/java",
"/etc/ld.so.cache", "/etc/localtime", "/etc/nsswitch.conf",
"/etc/passwd", "/etc/group", "/proc/self", "/proc/cpuinfo",
"/proc/meminfo", "/sys/devices/system/cpu"):
Expand Down
3 changes: 2 additions & 1 deletion tasks/financial-report-qa/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,8 @@ def child_env():
"HOME": "/work/index",
"USER": "submission",
"LOGNAME": "submission",
"PATH": "/opt/conda/bin:/usr/local/bin:/usr/bin:/bin",
"JAVA_HOME": "/opt/java/openjdk",
"PATH": "/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin",
"LANG": "C.UTF-8",
"PYTHONUNBUFFERED": "1",
"PYTHONDONTWRITEBYTECODE": "1",
Expand Down
4 changes: 3 additions & 1 deletion tasks/icsi-meeting-memory/tests/harness.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@
HERE = Path(__file__).resolve().parent
RUNTIME = [
"/opt/conda",
"/opt/java",
"/usr",
"/bin",
"/lib",
Expand Down Expand Up @@ -123,7 +124,8 @@ def invoke(argv, read, work, log, label, timeout, gateway=None):
work.mkdir(mode=0o700, exist_ok=True)
own(work, 10001, 10001)
env = {
"PATH": "/opt/conda/bin:/usr/local/bin:/usr/bin:/bin",
"JAVA_HOME": "/opt/java/openjdk",
"PATH": "/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin",
"HOME": str(work),
"TMPDIR": str(work),
"LANG": "C.UTF-8",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,8 @@ def prepare():
os.setuid(UID)
resource.setrlimit(resource.RLIMIT_CORE, (0, 0))
resource.setrlimit(resource.RLIMIT_FSIZE, (32 * 1024**3,) * 2)
env = {'PATH': '/opt/conda/bin:/usr/bin:/bin', 'HOME': str(state),
env = {'JAVA_HOME': '/opt/java/openjdk',
'PATH': '/opt/conda/bin:/opt/java/openjdk/bin:/usr/bin:/bin', 'HOME': str(state),
'TMPDIR': str(state), 'OMP_NUM_THREADS': '16',
'OPENBLAS_NUM_THREADS': '1', 'MKL_NUM_THREADS': '1',
'KMP_AFFINITY': 'disabled', 'OMP_PROC_BIND': 'false',
Expand Down
3 changes: 2 additions & 1 deletion tasks/incremental-vector-search/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -204,7 +204,8 @@ def monitor():
if peak_disk > 32 * GIB:
raise ValueError("Dataset exceeded 32 GiB storage")

env = {"PATH": "/opt/conda/bin:/usr/bin:/bin", "HOME": "/work/state",
env = {"JAVA_HOME": "/opt/java/openjdk",
"PATH": "/opt/conda/bin:/opt/java/openjdk/bin:/usr/bin:/bin", "HOME": "/work/state",
"TMPDIR": "/work/state", "OMP_NUM_THREADS": "16", "OPENBLAS_NUM_THREADS": "1",
"MKL_NUM_THREADS": "1", "KMP_AFFINITY": "disabled", "OMP_PROC_BIND": "false",
"PYTHONDONTWRITEBYTECODE": "1", "LC_ALL": "C.UTF-8"}
Expand Down
2 changes: 1 addition & 1 deletion tasks/incremental-vector-search/tests/isolate.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@


def main():
runtime = ["/usr", "/bin", "/lib", "/lib64", "/opt/conda", "/etc/ld.so.cache",
runtime = ["/usr", "/bin", "/lib", "/lib64", "/opt/conda", "/opt/java", "/etc/ld.so.cache",
"/etc/localtime", "/proc/cpuinfo", "/proc/meminfo", "/sys/devices/system/cpu"]
rules = [(p, READ | READ_DIR | EXECUTE) for p in runtime if Path(p).exists()]
rules += [("/app/submission", READ | READ_DIR | EXECUTE),
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ def __init__(self,role,output,resources):
self.work.chmod(0o700);os.chown(self.work,UID,UID)
self.log=open(output/f'{role}-worker.stderr','ab')
env={k:os.environ[k] for k in ['PATH','LD_LIBRARY_PATH','CUDA_VISIBLE_DEVICES','NVIDIA_VISIBLE_DEVICES','NVIDIA_DRIVER_CAPABILITIES'] if k in os.environ}
env.update(HOME=str(self.work),TMPDIR=str(self.work),ENCODER_WORK_DIR=str(self.work),XDG_CACHE_HOME=str(self.work/'cache'),HF_HOME=str(self.work/'hf'),
env.update(JAVA_HOME='/opt/java/openjdk', HOME=str(self.work),TMPDIR=str(self.work),ENCODER_WORK_DIR=str(self.work),XDG_CACHE_HOME=str(self.work/'cache'),HF_HOME=str(self.work/'hf'),
OMP_NUM_THREADS='16',OPENBLAS_NUM_THREADS='16',MKL_NUM_THREADS='16',TOKENIZERS_PARALLELISM='false',PYTHONUNBUFFERED='1',HF_HUB_OFFLINE='1',TRANSFORMERS_OFFLINE='1')
self.p=subprocess.Popen(['/opt/conda/bin/python','/task/runtime/worker.py',role],stdin=subprocess.PIPE,stdout=subprocess.PIPE,stderr=self.log,env=env,bufsize=0)
os.set_blocking(self.p.stdin.fileno(),False)
Expand Down
3 changes: 2 additions & 1 deletion tasks/join-aware-table-retrieval/tests/grade.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,8 @@
MAX_LINE = 65536
MAX_INDEX = 8 * 1024**3
ENV = {'HOME': '/home/submission', 'USER': 'submission', 'LOGNAME': 'submission',
'PATH': '/opt/conda/bin:/usr/local/bin:/usr/bin:/bin', 'LANG': 'C.UTF-8',
'JAVA_HOME': '/opt/java/openjdk',
'PATH': '/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin', 'LANG': 'C.UTF-8',
'PYTHONUNBUFFERED': '1', 'OMP_NUM_THREADS': '16', 'OPENBLAS_NUM_THREADS': '16',
'TOKENIZERS_PARALLELISM': 'false', 'TMPDIR': '/work/tmp'}

Expand Down
3 changes: 2 additions & 1 deletion tasks/khmer-embedding-training/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,8 @@ def run_stage(user, argv, work, log, timeout):
env = {
"HOME": str(work),
"USER": user,
"PATH": "/opt/conda/bin:/usr/local/bin:/usr/bin:/bin",
"JAVA_HOME": "/opt/java/openjdk",
"PATH": "/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin",
"LANG": "C.UTF-8",
"TMPDIR": str(work / "tmp"),
"HF_HUB_OFFLINE": "1",
Expand Down
3 changes: 2 additions & 1 deletion tasks/large-scale-formula-retrieval/tests/runtime.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,7 +116,8 @@ def sandbox(cpu_ids, work, allow_api=False):


def submission_env(work, cpu_ids, allow_api=False):
env = {'PATH': '/opt/conda/bin:/usr/local/bin:/usr/bin:/bin', 'HOME': str(work),
env = {'JAVA_HOME': '/opt/java/openjdk',
'PATH': '/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin', 'HOME': str(work),
'TMPDIR': str(work), 'XDG_CACHE_HOME': str(work), 'LANG': 'C.UTF-8',
'OMP_NUM_THREADS': str(len(cpu_ids)), 'OPENBLAS_NUM_THREADS': str(len(cpu_ids)), 'MKL_NUM_THREADS': str(len(cpu_ids)),
'OMP_PROC_BIND': 'false', 'KMP_AFFINITY': 'disabled',
Expand Down
3 changes: 2 additions & 1 deletion tasks/layout-page-search/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,8 @@ def child_env():
"HOME": "/work/index",
"USER": "submission",
"LOGNAME": "submission",
"PATH": "/opt/conda/bin:/usr/local/bin:/usr/bin:/bin",
"JAVA_HOME": "/opt/java/openjdk",
"PATH": "/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin",
"LANG": "C.UTF-8",
"PYTHONUNBUFFERED": "1",
"PYTHONDONTWRITEBYTECODE": "1",
Expand Down
3 changes: 2 additions & 1 deletion tasks/legal-stance-retrieval/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,8 @@ def submission_command(command: list[str]) -> list[str]:
f"HOME={INDEX_DIR}",
"USER=submission",
"LOGNAME=submission",
"PATH=/opt/conda/bin:/usr/local/bin:/usr/bin:/bin",
"JAVA_HOME=/opt/java/openjdk",
"PATH=/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin",
"LANG=C.UTF-8",
"PYTHONUNBUFFERED=1",
"PYTHONDONTWRITEBYTECODE=1",
Expand Down
3 changes: 2 additions & 1 deletion tasks/long-document-reranking/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,8 @@
"HOME=/home/submission",
"USER=submission",
"LOGNAME=submission",
"PATH=/opt/conda/bin:/usr/local/bin:/usr/bin:/bin",
"JAVA_HOME=/opt/java/openjdk",
"PATH=/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin",
"LANG=C.UTF-8",
"PYTHONUNBUFFERED=1",
]
Expand Down
3 changes: 2 additions & 1 deletion tasks/long-pdf-evidence-vectorless/tests/run_queries.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,8 @@
def submission_env():
env = {
"HOME": "/home/submission", "USER": "submission", "LOGNAME": "submission",
"PATH": "/opt/conda/bin:/usr/local/bin:/usr/bin:/bin", "LANG": "C.UTF-8",
"JAVA_HOME": "/opt/java/openjdk",
"PATH": "/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin", "LANG": "C.UTF-8",
"PYTHONUNBUFFERED": "1", "TOKENIZERS_PARALLELISM": "false",
}
for key in ("OPENROUTER_API_KEY", "HTTP_PROXY", "HTTPS_PROXY",
Expand Down
3 changes: 2 additions & 1 deletion tasks/long-pdf-evidence-vectorless/tests/test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,8 @@ SUBMISSION_COMMAND=(
"HOME=/home/submission"
"USER=submission"
"LOGNAME=submission"
"PATH=/opt/conda/bin:/usr/local/bin:/usr/bin:/bin"
"JAVA_HOME=/opt/java/openjdk"
"PATH=/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin"
"LANG=C.UTF-8"
"PYTHONUNBUFFERED=1"
"TOKENIZERS_PARALLELISM=false"
Expand Down
3 changes: 2 additions & 1 deletion tasks/long-pdf-evidence/tests/run_queries.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,8 @@
def submission_env():
env = {
"HOME": "/home/submission", "USER": "submission", "LOGNAME": "submission",
"PATH": "/opt/conda/bin:/usr/local/bin:/usr/bin:/bin", "LANG": "C.UTF-8",
"JAVA_HOME": "/opt/java/openjdk",
"PATH": "/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin", "LANG": "C.UTF-8",
"PYTHONUNBUFFERED": "1", "TOKENIZERS_PARALLELISM": "false",
}
for key in ("OPENROUTER_API_KEY", "JINA_API_KEY", "HTTP_PROXY", "HTTPS_PROXY",
Expand Down
3 changes: 2 additions & 1 deletion tasks/long-pdf-evidence/tests/test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,8 @@ SUBMISSION_COMMAND=(
"HOME=/home/submission"
"USER=submission"
"LOGNAME=submission"
"PATH=/opt/conda/bin:/usr/local/bin:/usr/bin:/bin"
"JAVA_HOME=/opt/java/openjdk"
"PATH=/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin"
"LANG=C.UTF-8"
"PYTHONUNBUFFERED=1"
"TOKENIZERS_PARALLELISM=false"
Expand Down
3 changes: 2 additions & 1 deletion tasks/memory-constrained-dense-retrieval/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,8 @@
f"HOME={INDEX_DIR}",
"USER=submission",
"LOGNAME=submission",
"PATH=/opt/conda/bin:/usr/local/bin:/usr/bin:/bin",
"JAVA_HOME=/opt/java/openjdk",
"PATH=/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin",
"LANG=C.UTF-8",
"PYTHONUNBUFFERED=1",
"PYTHONDONTWRITEBYTECODE=1",
Expand Down
3 changes: 2 additions & 1 deletion tasks/methodology-inspiration-retrieval/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,8 @@ def read_jsonl(path):


def child_command(command):
environment = ["PATH=/opt/conda/bin:/usr/local/bin:/usr/bin:/bin", "USER=submission",
environment = ["JAVA_HOME=/opt/java/openjdk",
"PATH=/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin", "USER=submission",
"LOGNAME=submission", f"HOME={INDEX}", "LANG=C.UTF-8", "PYTHONUNBUFFERED=1",
"PYTHONDONTWRITEBYTECODE=1", "TOKENIZERS_PARALLELISM=false",
"OMP_NUM_THREADS=8", "OPENBLAS_NUM_THREADS=8", "HF_HUB_OFFLINE=1",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ def __init__(self, submission, work, log, deadline, sandbox=False, require_gpu=T
work.mkdir(parents=True, exist_ok=True)
work.chmod(0o777)
environment = {k: os.environ[k] for k in ("PATH", "LD_LIBRARY_PATH", "CUDA_VISIBLE_DEVICES", "NVIDIA_VISIBLE_DEVICES", "NVIDIA_DRIVER_CAPABILITIES", "SEARCH_SWE_VERIFIER_SANDBOX") if k in os.environ}
environment.update(HOME=str(work), TMPDIR=str(work), XDG_CACHE_HOME=str(work / "cache"),
environment.update(JAVA_HOME="/opt/java/openjdk", HOME=str(work), TMPDIR=str(work), XDG_CACHE_HOME=str(work / "cache"),
CUDA_CACHE_PATH=str(work / "cuda-cache"), TRITON_CACHE_DIR=str(work / "triton-cache"),
OMP_NUM_THREADS="16", OPENBLAS_NUM_THREADS="16", MKL_NUM_THREADS="16",
PYTHONNOUSERSITE="1", PYTHONDONTWRITEBYTECODE="1", HF_HUB_OFFLINE="1", TRANSFORMERS_OFFLINE="1")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ def __init__(self, submission, work, log, deadline, sandbox=False, require_gpu=T
work.mkdir(parents=True, exist_ok=True)
work.chmod(0o777)
environment = {k: os.environ[k] for k in ("PATH", "LD_LIBRARY_PATH", "CUDA_VISIBLE_DEVICES", "NVIDIA_VISIBLE_DEVICES", "NVIDIA_DRIVER_CAPABILITIES", "SEARCH_SWE_VERIFIER_SANDBOX") if k in os.environ}
environment.update(HOME=str(work), TMPDIR=str(work), XDG_CACHE_HOME=str(work / "cache"),
environment.update(JAVA_HOME="/opt/java/openjdk", HOME=str(work), TMPDIR=str(work), XDG_CACHE_HOME=str(work / "cache"),
CUDA_CACHE_PATH=str(work / "cuda-cache"), TRITON_CACHE_DIR=str(work / "triton-cache"),
OMP_NUM_THREADS="16", OPENBLAS_NUM_THREADS="16", MKL_NUM_THREADS="16",
PYTHONNOUSERSITE="1", PYTHONDONTWRITEBYTECODE="1", HF_HUB_OFFLINE="1", TRANSFORMERS_OFFLINE="1")
Expand Down
3 changes: 2 additions & 1 deletion tasks/narrative-clue-search/tests/grader.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,8 @@ def child_env():
"HOME": "/work/index",
"USER": "submission",
"LOGNAME": "submission",
"PATH": "/opt/conda/bin:/usr/local/bin:/usr/bin:/bin",
"JAVA_HOME": "/opt/java/openjdk",
"PATH": "/opt/conda/bin:/opt/java/openjdk/bin:/usr/local/bin:/usr/bin:/bin",
"LANG": "C.UTF-8",
"PYTHONUNBUFFERED": "1",
"PYTHONDONTWRITEBYTECODE": "1",
Expand Down
Loading
Loading