diff --git a/environment.yml b/environment.yml index f85ad05..6c9fc28 100644 --- a/environment.yml +++ b/environment.yml @@ -41,13 +41,11 @@ dependencies: - mkl_random=1.1.1=py36h0573a6f_0 - ncurses=6.3=h5eee18b_3 - nettle=3.7.3=hbbd107a_1 - - numpy=1.19.2=py36h54aff64_0 - numpy-base=1.19.2=py36hfa32c7d_0 - olefile=0.46=py36_0 - openh264=2.1.1=h4ff587b_0 - openjpeg=2.4.0=h3ad879b_0 - openssl=1.1.1q=h7f8727e_0 - - pillow=8.3.1=py36h2c7a002_0 - pip=21.2.2=py36h06a4308_0 - python=3.6.13=h12debd9_1 - pytorch=1.10.2=py3.6_cuda11.3_cudnn8.2.0_0 @@ -75,7 +73,7 @@ dependencies: - click==8.0.4 - datasets==2.4.0 - dill==0.3.4 - - easy-elasticsearch==0.0.8 + - easy-elasticsearch==0.0.9 - elasticsearch==7.9.1 - faiss-cpu==1.7.2 - filelock==3.4.1 @@ -89,8 +87,10 @@ dependencies: - multidict==5.2.0 - multiprocess==0.70.12.2 - nltk==3.6.7 + - numpy==1.19.5 - packaging==21.3 - pandas==1.1.5 + - pillow==8.4.0 - protobuf==3.19.5 - pyarrow==6.0.1 - pyparsing==3.0.9 diff --git a/setup.py b/setup.py index beaa93b..26e373c 100644 --- a/setup.py +++ b/setup.py @@ -6,7 +6,7 @@ setup( name="gpl", - version="0.1.2", + version="0.1.3", author="Kexin Wang", author_email="kexin.wang.2049@gmail.com", description="GPL is an unsupervised domain adaptation method for training dense retrievers. It is based on query generation and pseudo labeling with powerful cross-encoders. To train a domain-adapted model, it needs only the unlabeled target corpus and can achieve significant improvement over zero-shot models.", @@ -23,5 +23,5 @@ "Operating System :: OS Independent", ], python_requires=">=3.6", - install_requires=["beir", "easy-elasticsearch>=0.0.8", "protobuf"], + install_requires=["beir", "easy-elasticsearch>=0.0.9", "protobuf"], )