add typehints
Browse files- src/axolotl/prompters.py +2 -2
src/axolotl/prompters.py
CHANGED
|
@@ -266,12 +266,12 @@ class ShareGPTPrompter: # pylint: disable=too-few-public-methods
|
|
| 266 |
A prompter that generates prompts for the ShareGPT
|
| 267 |
"""
|
| 268 |
|
| 269 |
-
def __init__(self, prompt_style=None, system_prompt=None):
|
| 270 |
if prompt_style != PromptStyle.CHAT.value:
|
| 271 |
raise ValueError(
|
| 272 |
f"unsupported prompt_style for ShareGPTPrompter({prompt_style})"
|
| 273 |
)
|
| 274 |
-
system = (
|
| 275 |
system_prompt
|
| 276 |
if system_prompt
|
| 277 |
else (
|
|
|
|
| 266 |
A prompter that generates prompts for the ShareGPT
|
| 267 |
"""
|
| 268 |
|
| 269 |
+
def __init__(self, prompt_style=None, system_prompt: Optional[str] = None):
|
| 270 |
if prompt_style != PromptStyle.CHAT.value:
|
| 271 |
raise ValueError(
|
| 272 |
f"unsupported prompt_style for ShareGPTPrompter({prompt_style})"
|
| 273 |
)
|
| 274 |
+
system: str = (
|
| 275 |
system_prompt
|
| 276 |
if system_prompt
|
| 277 |
else (
|