iohex/cybergym
1
0
Fork 0
mirror of https://github.com/sunblaze-ucb/cybergym.git synced 2026-10-12 08:42:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix download scripts

Browse Source
This commit is contained in:
stneng
2026-02-09 22:20:38 -08:00
parent 1ba73e2842
commit 1442621830
3 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/server_data/download.py
View File

@@ -1,8 +1,8 @@
import argparse import argparse
import sys
from concurrent.futures import ThreadPoolExecutor, as_completed
import docker
import json import json
from concurrent.futures import ThreadPoolExecutor, as_completed
import docker
client = docker.from_env() client = docker.from_env()
@@ -53,7 +53,7 @@ if __name__ == "__main__":
arvo_id = task["task_id"].split(":")[-1] arvo_id = task["task_id"].split(":")[-1]
tags_arvo.append(f"{arvo_id}-vul") tags_arvo.append(f"{arvo_id}-vul")
tags_arvo.append(f"{arvo_id}-fix") tags_arvo.append(f"{arvo_id}-fix")
if task["task_id"].split(":")[0] == "ossfuzz": if task["task_id"].split(":")[0] == "oss-fuzz":
ossfuzz_id = task["task_id"].split(":")[-1] ossfuzz_id = task["task_id"].split(":")[-1]
tags_ossfuzz.append(f"{ossfuzz_id}-vul") tags_ossfuzz.append(f"{ossfuzz_id}-vul")
tags_ossfuzz.append(f"{ossfuzz_id}-fix") tags_ossfuzz.append(f"{ossfuzz_id}-fix")

4
scripts/server_data/download_binary_only_runners.py
View File

@@ -1,8 +1,6 @@
import argparse
import sys
from concurrent.futures import ThreadPoolExecutor, as_completed from concurrent.futures import ThreadPoolExecutor, as_completed
import docker import docker
import json
client = docker.from_env() client = docker.from_env()

3
scripts/server_data/download_subset.py
View File

@@ -1,8 +1,7 @@
import argparse import argparse
import sys
from concurrent.futures import ThreadPoolExecutor, as_completed from concurrent.futures import ThreadPoolExecutor, as_completed
import docker import docker
import json
client = docker.from_env() client = docker.from_env()