Fix args.tasks type check issue

duanshengliu · duanshengliu · commit 0b0071940352 · 2024-07-23T21:16:17.000+08:00
Signed-off-by: duansheng.liu &lt;1069033158@qq.com&gt;
diff --git a/intel_extension_for_transformers/transformers/llm/evaluation/lm_eval/accuracy.py b/intel_extension_for_transformers/transformers/llm/evaluation/lm_eval/accuracy.py
@@ -89,7 +89,7 @@ def cli_evaluate(args) -> None:
     if args.tasks is None:
         eval_logger.error("Need to specify task to evaluate.")
         sys.exit()
-    elif args.tasks == "list":
+    elif isinstance(args.tasks, list):
         eval_logger.info(
             "Available Tasks:\n - {}".format("\n - ".join(task_manager.all_tasks))
         )

Original file line number	Diff line number	Diff line change
`@@ -89,7 +89,7 @@ def cli_evaluate(args) -> None:`
`89`	`89`	`if args.tasks is None:`
`90`	`90`	`eval_logger.error("Need to specify task to evaluate.")`
`91`	`91`	`sys.exit()`
`92`		`- elif args.tasks == "list":`
	`92`	`+ elif isinstance(args.tasks, list):`
`93`	`93`	`eval_logger.info(`
`94`	`94`	`"Available Tasks:\n - {}".format("\n - ".join(task_manager.all_tasks))`
`95`	`95`	`)`