diff --git a/fastchat/serve/monitor/classify/label.py b/fastchat/serve/monitor/classify/label.py index 1ae42b731..fe928a9ba 100644 --- a/fastchat/serve/monitor/classify/label.py +++ b/fastchat/serve/monitor/classify/label.py @@ -399,6 +399,7 @@ def find_required_tasks(row): if category.name_tag in row["required_tasks"] ], args.testing, + config["api_type"], ) futures.append(future) for future in tqdm.tqdm(