From 81d426a17719877b45fb46c933ae6c3d2b99cf10 Mon Sep 17 00:00:00 2001 From: Ubuntu Date: Fri, 31 May 2024 20:38:53 +0000 Subject: [PATCH] format --- llmfoundry/utils/exceptions.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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."""