diff --git a/llmfoundry/utils/exceptions.py b/llmfoundry/utils/exceptions.py index ac068f87d4..e7fe0dd6ea 100644 --- a/llmfoundry/utils/exceptions.py +++ b/llmfoundry/utils/exceptions.py @@ -172,13 +172,13 @@ def __init__(self, example_keys: str) -> None: super().__init__(message, example_keys=example_keys) + class ExampleDatasetKeyCaseError(UserError): """Error thrown when keys in a dataset example are not in lowercase. This error checks for keys that could potentially match the expected example types if corrected. """ - def __init__(self, example_keys: str) -> None: message = ( f"Found keys {example_keys} in the dataset. All keys in datasets must be in lowercase. " @@ -186,6 +186,7 @@ def __init__(self, example_keys: str) -> None: ) super().__init__(message, example_keys=example_keys) + class NotEnoughChatDataError(UserError): """Error thrown when there is not enough chat data to train a model."""