winglian commited on
Commit
c7dee56
·
1 Parent(s): aac4b76

add typehints

Browse files
Files changed (1) hide show
  1. 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 (