Skip to content

Commit 165fc8c

Browse files
committed
🐧 use older fairseq version for MacOS users
1 parent c79da1b commit 165fc8c

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

setup.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
from setuptools import setup, find_packages
2+
import sys
23

3-
VERSION = '0.5.3'
4+
VERSION = '0.5.4'
45

56
install_requires = [
67
"torch==1.13.1",
78
'numpy<=1.23.5',
89
'ipython<=8.12', # python 3.8 vs ipython 8.13 incompatibility
910
'tqdm>=4.55.0',
10-
'fairseq==0.10.2', # seems to not work on python 3.9 for some platforms
1111
'packaging>=20.8',
1212
'requests>=2.25.1',
1313
# for data (ELMO embeddings)
@@ -26,6 +26,14 @@
2626
'progressbar',
2727
]
2828

29+
if sys.platform == 'darwin':
30+
install_requires.append("fairseq==0.10.0")
31+
# torch-scatter can be directly installed on macos;
32+
# it needs to be separately install with url on linux
33+
install_requires.append("torch-scatter")
34+
else:
35+
install_requires.append("fairseq==0.10.2")
36+
2937
if __name__ == '__main__':
3038
setup(
3139
name='transition_amr_parser',

0 commit comments

Comments
 (0)