From 5f1418a68b3757ef798363267efb29c0ffb22601 Mon Sep 17 00:00:00 2001 From: codingma Date: Tue, 16 Apr 2024 10:56:39 +0800 Subject: [PATCH] add check Former-commit-id: 008f6498977c243c80e87242f05c9cf9573541ac --- src/llmtuner/webui/components/data.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/src/llmtuner/webui/components/data.py b/src/llmtuner/webui/components/data.py index c0f113ea..ab6b5de4 100644 --- a/src/llmtuner/webui/components/data.py +++ b/src/llmtuner/webui/components/data.py @@ -28,14 +28,13 @@ def can_preview(dataset_dir: str, dataset: list) -> "gr.Button": dataset_info = json.load(f) except Exception: return gr.Button(interactive=False) - + + if len(dataset) == 0 or "file_name" not in dataset_info[dataset[0]]: + return gr.Button(interactive=False) + local_path = os.path.join(dataset_dir, dataset_info[dataset[0]]["file_name"]) - if ( - len(dataset) > 0 - and "file_name" in dataset_info[dataset[0]] - and (os.path.isfile(local_path) - or (os.path.isdir(local_path)) and len(os.listdir(local_path)) != 0) - ): + if (os.path.isfile(local_path) + or (os.path.isdir(local_path) and len(os.listdir(local_path)) != 0)): return gr.Button(interactive=True) else: return gr.Button(interactive=False)