Merge branch 'main' into final-register-only

mosaicml · Sep 11, 2024 · 2f0db9c · 2f0db9c
2 parents 44d41da + 6849eaa
commit 2f0db9c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/data/finetuning/tasks.py b/llmfoundry/data/finetuning/tasks.py
@@ -1013,7 +1013,7 @@ def dataset_mapper(example: dict):
             raise MisconfiguredHfDatasetError(
                 dataset_name=dataset_name,
                 split=split,
-            )
+            ) from error
         if error is not None:
             log.error('Error during data prep')
             raise error