update data

microsoft · Sep 23, 2024 · 342e3b8 · 342e3b8
1 parent 17b63b6
commit 342e3b8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/taskbench/inference.py b/taskbench/inference.py
@@ -36,6 +36,9 @@ def __init__(self, message):
 @click.option("--log_first_detail", type=bool, default=False)
 def main(data_dir, temperature, top_p, api_addr, api_key, api_port, multiworker, llm, use_demos, reformat, reformat_by, tag, dependency_type, log_first_detail):
     assert dependency_type in ["resource", "temporal"], "Dependency type not supported"
+    if dependency_type == "resource":
+        assert data_dir != "data_dailylifeapis", "Resource dependency type only support data_huggingface and data_multimedia"
+
     arguments = locals()
     url = f"http://{api_addr}:{api_port}/v1/chat/completions"
     header = {