Skip to content

Commit 4b914cb

Browse files
committed
Merge branch 'main' of github.com:apmoore1/stanza-batch into main
2 parents 9245858 + 9ac31d1 commit 4b914cb

File tree

1 file changed

+12
-10
lines changed

1 file changed

+12
-10
lines changed

stanza_batch/__init__.py

Lines changed: 12 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -158,11 +158,12 @@ def change_offsets(
158158
sentence_dicts: List[Dict[str, str]] = []
159159
sentence_sentiment = getattr(sentence, "sentiment", None)
160160
for token in sentence.to_dict():
161-
token_misc = token["misc"]
162-
start, end = _start_end_character_offsets(token_misc)
163-
start = start - start_offset
164-
end = end - start_offset
165-
token["misc"] = f"start_char={start}|end_char={end}"
161+
if "misc" in token:
162+
token_misc = token["misc"]
163+
start, end = _start_end_character_offsets(token_misc)
164+
start = start - start_offset
165+
end = end - start_offset
166+
token["misc"] = f"start_char={start}|end_char={end}"
166167
token["sentence_sentiment"] = sentence_sentiment
167168
sentence_dicts.append(token)
168169
all_sentence_dicts.append(sentence_dicts)
@@ -263,11 +264,12 @@ def change_offsets(
263264
sentence_dicts: List[Dict[str, str]] = []
264265
sentence_sentiment = getattr(sentence, "sentiment", None)
265266
for token in sentence.to_dict():
266-
token_misc = token["misc"]
267-
start, end = _start_end_character_offsets(token_misc)
268-
start = start + offset_to_add
269-
end = end + offset_to_add
270-
token["misc"] = f"start_char={start}|end_char={end}"
267+
if "misc" in token:
268+
token_misc = token["misc"]
269+
start, end = _start_end_character_offsets(token_misc)
270+
start = start + offset_to_add
271+
end = end + offset_to_add
272+
token["misc"] = f"start_char={start}|end_char={end}"
271273
token["sentence_sentiment"] = sentence_sentiment
272274
sentence_dicts.append(token)
273275
all_sentence_dicts.append(sentence_dicts)

0 commit comments

Comments
 (0)