Update typing

- Update typing
- Change my e-mail

Author: wannaphong

pythainlp/corpus/default_db.json

@@ -8,7 +8,7 @@
         "authors": [
             "Wannaphong Phatthiyaphaibun"
         ],
-        "author_email": "wannaphong@kkumail.com",
+        "author_email": "wannaphong@yahoo.com",
         "license": "cc-by-4.0",
         "versions": {
             "1.5.1": {

pythainlp/soundex/sound.py

@@ -44,7 +44,7 @@ def word2audio(word: str) -> str:
     _ipa = [_clean_ipa(transliterate(phone, engine="thaig2p")) for phone in _phone]
     return '.'.join(_ipa)
 
-def audio_vector(word:str) -> List[List[int]]:
+def audio_vector(word: str) -> List[List[int]]:
     """
     Convert audio to vector list
 
@@ -62,14 +62,14 @@ def audio_vector(word:str) -> List[List[int]]:
     """
     return _ft.word_to_vector_list(word2audio(word), numeric=True)
 
-def word_approximation(word:str, list_word:List[str]):
+def word_approximation(word: str, list_word: List[str]) -> List[float]:
     """
     Thai Word Approximation
 
     :param str word: Thai word
     :param str list_word: Thai word
     :return: List of approximation of words (The smaller the value, the closer)
-    :rtype: List[str]
+    :rtype: List[float]
 
     :Example:
     ::

pythainlp/spell/wanchanberta_thai_grammarly.py

@@ -11,6 +11,7 @@
     * GitHub: \
         https://github.com/bookpanda/Two-stage-Thai-Misspelling-Correction-Based-on-Pre-trained-Language-Models
 """
+from typing import List
 import torch
 from transformers import (
     AutoModelForMaskedLM,
@@ -36,7 +37,7 @@ def forward(self, input_id, mask, label):
     tagging_model = tagging_model.to(device=device)
 ids_to_labels = {0: 'f', 1: 'i'}
 
-def align_word_ids(texts):
+def align_word_ids(texts: str) -> List[int]:
     tokenized_inputs = tokenizer(texts, padding='max_length', max_length=512, truncation=True)
     word_ids = tokenized_inputs.word_ids()
     label_ids = []
@@ -70,7 +71,7 @@ def evaluate_one_text(model, sentence):
 if use_cuda:
     mlm_model = mlm_model.to(device=device)
 
-def correct(text):
+def correct(text: str) -> str:
     ans = []
     i_f = evaluate_one_text(tagging_model, text)
     a = tokenizer(text)

pythainlp/tag/_tag_perceptron.py

@@ -8,7 +8,7 @@
 This tagger is a port of the Textblob Averaged Perceptron Tagger
 Author: Matthew Honnibal <[email protected]>,
         Long Duong <[email protected]> (NLTK port)
-        Wannaphong Phatthiyaphaibun <wannaphong@kkumail.com> (PyThaiNLP port)
+        Wannaphong Phatthiyaphaibun <wannaphong@yahoo.com> (PyThaiNLP port)
 URL: <https://github.com/sloria/textblob-aptagger>
      <https://nltk.org/>
 Copyright 2013 Matthew Honnibal