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 (
|