diff --git a/scripts/server_data/download.py b/scripts/server_data/download.py index 769629c..da55581 100644 --- a/scripts/server_data/download.py +++ b/scripts/server_data/download.py @@ -1,8 +1,8 @@ import argparse -import sys -from concurrent.futures import ThreadPoolExecutor, as_completed -import docker import json +from concurrent.futures import ThreadPoolExecutor, as_completed + +import docker client = docker.from_env() @@ -53,7 +53,7 @@ if __name__ == "__main__": arvo_id = task["task_id"].split(":")[-1] tags_arvo.append(f"{arvo_id}-vul") 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] tags_ossfuzz.append(f"{ossfuzz_id}-vul") tags_ossfuzz.append(f"{ossfuzz_id}-fix") diff --git a/scripts/server_data/download_binary_only_runners.py b/scripts/server_data/download_binary_only_runners.py index e22d309..6f43e7a 100644 --- a/scripts/server_data/download_binary_only_runners.py +++ b/scripts/server_data/download_binary_only_runners.py @@ -1,8 +1,6 @@ -import argparse -import sys from concurrent.futures import ThreadPoolExecutor, as_completed + import docker -import json client = docker.from_env() diff --git a/scripts/server_data/download_subset.py b/scripts/server_data/download_subset.py index fd45151..81359ea 100644 --- a/scripts/server_data/download_subset.py +++ b/scripts/server_data/download_subset.py @@ -1,8 +1,7 @@ import argparse -import sys from concurrent.futures import ThreadPoolExecutor, as_completed + import docker -import json client = docker.from_env()