Skip to content

Commit 7147fe8

Browse files
committed
Raise errors when instance is not ready
1 parent 6b72bde commit 7147fe8

3 files changed

Lines changed: 11 additions & 0 deletions

File tree

pythainlp/tokenize/budoux.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,9 @@ def segment(text: str) -> list[str]:
5656
_parser = _init_parser()
5757
parser = _parser
5858

59+
if parser is None:
60+
raise RuntimeError("Failed to initialize BudouX parser")
61+
5962
result = cast("list[str]", parser.parse(text))
6063

6164
return result

pythainlp/transliterate/thai2rom.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -245,6 +245,10 @@ def forward(
245245
attn_energies,
246246
self.other.unsqueeze(0).expand(*hidden.size()).transpose(1, 2),
247247
).squeeze(2)
248+
else:
249+
raise ValueError(
250+
f"Unsupported attention method: {self.method!r}"
251+
)
248252

249253
attn_energies = attn_energies.masked_fill(mask == 0, -1e10)
250254

pythainlp/transliterate/thaig2p.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -267,6 +267,10 @@ def forward(
267267
attn_energies,
268268
self.other.unsqueeze(0).expand(*hidden.size()).transpose(1, 2),
269269
).squeeze(2)
270+
else:
271+
raise ValueError(
272+
f"Unsupported attention method: {self.method!r}"
273+
)
270274

271275
attn_energies = attn_energies.masked_fill(mask == 0, -1e10)
272276

0 commit comments

Comments
 (0)