Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
Ubuntu committed May 31, 2024
1 parent 42fcd8d commit 81d426a
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion llmfoundry/utils/exceptions.py
Original file line number Diff line number Diff line change
Expand Up @@ -172,20 +172,21 @@ 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. "
f"Please ensure all keys are formatted correctly."
)
super().__init__(message, example_keys=example_keys)


class NotEnoughChatDataError(UserError):
"""Error thrown when there is not enough chat data to train a model."""

Expand Down

0 comments on commit 81d426a

Please sign in to comment.