Merge pull request #111 from amir-zeldes/develop

update requirements

requirements.txt

@@ -1,4 +1,7 @@
 scikit-learn>=0.22.1
 xgboost>=0.81,<=0.90
 flair==0.6.1
-xmltodict
+xmltodict
+transformers==3.0.2
+numpy>=1.20.0,<=1.24.4
+scikit-learn==0.22.1

setup.py

@@ -3,15 +3,15 @@
 setup(
   name = 'xrenner',
   packages = find_packages(),
-  version = '2.2.0.0',
+  version = '2.2.1.1',
   description = 'A configurable, language independent coreferencer and (non) named entity recognizer',
   author = 'Amir Zeldes',
   author_email = '[email protected]',
   package_data = {'':['README.rst','LICENSE.txt'],'xrenner':['models/*','test/*','licenses/*','models/_sequence_taggers/*']},
-   install_requires=['scikit-learn>=0.22.1','xgboost==0.90','flair==0.6.1','xmltodict'],
+   install_requires=['scikit-learn>=0.22.1','xgboost==0.90','flair==0.6.1','xmltodict','transformers==3.0.2','numpy>=1.20.0,<=1.24.4','scikit-learn==0.22.1'],
   url = 'https://github.com/amir-zeldes/xrenner',
   license='Apache License, Version 2.0',
-  download_url = 'https://github.com/amir-zeldes/xrenner/releases/tag/v2.2.0.0',
+  download_url = 'https://github.com/amir-zeldes/xrenner/releases/tag/v2.2.1.1',
   keywords = ['NLP', 'coreference', 'NER', 'named entity'],
   classifiers = ['Programming Language :: Python',
 'Programming Language :: Python :: 2',

xrenner/modules/get_models.py

@@ -25,7 +25,7 @@ def check_models(path=None):
 		model = path.split(os.sep)[-1]
 
 	if not os.path.exists(path):
-		server = "corpling.uis.georgetown.edu"
+		server = "gucorpling.org"
 		resource = "/".join([server, "amir", "download", model])
 		download_file("https://" + resource, os.sep.join([script_dir,"..","models","_sequence_taggers",model]))
 	else:

xrenner/modules/xrenner_out.py

@@ -184,6 +184,8 @@ def output_conll_sent(conll_tokens, markstart_dict, markend_dict, file_name, out
 					coref_col += ")"
 		if int(out_tok.id) not in markstart_dict and int(out_tok.id) not in markend_dict:
 			coref_col = "_"
+		if coref_col == "":
+			coref_col = "_"
 
 		line += infstat_col + "\t" + coref_col
 		output_string += line + "\n"

xrenner/xrenner.py

@@ -15,7 +15,7 @@
 import ctypes
 from math import ceil
 
-__version__ = "2.2.0"
+__version__ = "2.2.1"
 xrenner_version = "xrenner V" + __version__
 
 sys.dont_write_bytecode = True