Merge branch 'master' of https://github.com/dbklim/StressRNN

dbklim · Aug 7, 2024 · 0afefac · 0afefac
2 parents c3fee0a + 63aac81
commit 0afefac
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -178,7 +178,7 @@ python3 convert_zaliznyak_to_exception_dictionary.py -iz odict.zip -o stressrnn/
 
 ## Datasets
 
-The repo [contains](https://github.com/Desklop/StressRNN/datasets) samples from UD treebanks annotated with word stress for the Russian, Ukranian and Belorusian languages. For more details about the tool see VarDial paper (coming soon).
+The repo [contains](https://github.com/Desklop/StressRNN/tree/master/datasets) samples from UD treebanks annotated with word stress for the Russian, Ukranian and Belorusian languages. For more details about the tool see VarDial paper (coming soon).
 
 ## Authors
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
 numpy>=1.16.0
 h5py>=3.9.0
-pymorphy2[fast]>=0.9.1
+pymorphy2[fast]@git+https://github.com/pymorphy2-fork/pymorphy2@master
 onnxruntime>=1.15.1
diff --git a/requirements_dev.txt b/requirements_dev.txt
@@ -1,7 +1,7 @@
 numpy>=1.16.0
 h5py>=3.9.0
 protobuf<=3.20.3
-pymorphy2[fast]>=0.9.1
+pymorphy2[fast] @ git+https://github.com/pymorphy2-fork/pymorphy2@master
 tensorflow<=2.2.2
 onnxruntime>=1.15.1
 keras2onnx<=1.7
diff --git a/setup.py b/setup.py
@@ -34,6 +34,8 @@ def requirements() -> List[str]:
       keywords='nlp russian stress accent emphasis linguistic rnn lstm bilstm',
       author_email='ponomarevamawa@gmail.com (RusStress), vladsklim@gmail.com (StressRNN)',
       packages=['stressrnn'],
+      # Add py.typed to the package
+      package_data={'stressrnn': ['py.typed']},
       install_requires=requirements(),
       include_package_data=True,
       zip_safe=False)

diff --git a/stressrnn/exception_dictionary_wrapper.py b/stressrnn/exception_dictionary_wrapper.py
@@ -51,7 +51,7 @@ def __load_exception_dict(self, f_name_exception_dict: str, overwrite: bool = Fa
         several identical words with different stresses, they will all be added to the dictionary, and the stress positions will be
         specified in the order of reading the words. '''
 
-        with open(f_name_exception_dict, 'r') as f_exception_dict:
+        with open(f_name_exception_dict, 'r', encoding="utf-8") as f_exception_dict:
             for word in f_exception_dict:
                 word = word.strip('\n')
                 if DEF_STRESS_SYMBOL in word:

diff --git a/stressrnn/py.typed b/stressrnn/py.typed