From 64fc621d44440445a1a90949c9c135e53c0d7c97 Mon Sep 17 00:00:00 2001 From: Niels Geens Date: Fri, 18 Apr 2025 12:42:24 +0000 Subject: [PATCH] Fix error print in dataset --- tools/train/train/dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/train/train/dataset.py b/tools/train/train/dataset.py index 0562eec..c558ef4 100644 --- a/tools/train/train/dataset.py +++ b/tools/train/train/dataset.py @@ -120,12 +120,12 @@ class Dataset(torch.utils.data.Dataset): # Check system prompt hash file_system_hash = root.get('system_prompt_hash') if file_system_hash != self.system_prompt_hash: - print(f"WARNING: System prompt hash mismatch in {file_path}") + print(f"WARNING: System prompt hash mismatch in {file}") # Check action schema hash file_schema_hash = root.get('action_schema_hash') if file_schema_hash != self.action_schema_hash: - print(f"WARNING: Action schema hash mismatch in {file_path}") + print(f"WARNING: Action schema hash mismatch in {file}") # Check for required elements context_elem = root.find('context')