Skip to content

Commit

Permalink
style: reformat
Browse files Browse the repository at this point in the history
  • Loading branch information
wochinge committed Mar 6, 2024
1 parent 6b8118b commit 8ff9e8b
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@

class SagemakerError(Exception):
"""
Parent class for all exceptions raised by the Sagemaker component
"""


class AWSConfigurationError(SagemakerError):
"""Exception raised when AWS is not configured correctly"""

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,7 @@ class SagemakerGenerator:
>>> and respond to natural human language in a way that is both meaningful and useful.'], 'meta': [{}]}
```
"""

model_generation_keys: ClassVar = ["generated_text", "generation"]

def __init__(
Expand Down

0 comments on commit 8ff9e8b

Please sign in to comment.