aadnk commited on
Commit
f641aea
1 Parent(s): 74b7d77

Use type hints instead

Browse files
Files changed (1) hide show
  1. src/prompts/jsonPromptStrategy.py +2 -1
src/prompts/jsonPromptStrategy.py CHANGED
@@ -1,4 +1,5 @@
1
  import json
 
2
  from src.prompts.abstractPromptStrategy import AbstractPromptStrategy
3
 
4
 
@@ -25,7 +26,7 @@ class JsonPromptStrategy(AbstractPromptStrategy):
25
 
26
  """
27
  parsed_json = json.loads(initial_json_prompt)
28
- self.segment_lookup = dict[str, JsonPromptSegment]()
29
 
30
  for prompt_entry in parsed_json:
31
  segment_index = prompt_entry["segment_index"]
1
  import json
2
+ from typing import Dict
3
  from src.prompts.abstractPromptStrategy import AbstractPromptStrategy
4
 
5
 
26
 
27
  """
28
  parsed_json = json.loads(initial_json_prompt)
29
+ self.segment_lookup: Dict[str, JsonPromptSegment] = dict()
30
 
31
  for prompt_entry in parsed_json:
32
  segment_index = prompt_entry["segment_index"]