diff --git a/requirements.txt b/requirements.txt index 678f736fdec050ab84c8f7a3f04b53cf8f54ba8c..3e2b8960210d9c67dcf2b0698e10221fc3c8930c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,15 +1,14 @@ +#for python2, you should use requirements_py2.txt pre-commit protobuf >= 3.1.0 yapf == 0.26.0 pyyaml numpy -#[py2]numpy == 1.16.0 Pillow six >= 1.10.0 chardet == 3.0.4 requests pandas -#[py2]pandas == 0.24.0 flake8 tb-paddle tb-nightly diff --git a/requirements_py2.txt b/requirements_py2.txt new file mode 100644 index 0000000000000000000000000000000000000000..3557a744cedd5242f016a9df58de494994dda86d --- /dev/null +++ b/requirements_py2.txt @@ -0,0 +1,14 @@ +pre-commit +protobuf >= 3.1.0 +yapf == 0.26.0 +pyyaml +numpy < 1.17.0 +Pillow +six >= 1.10.0 +chardet == 3.0.4 +requests +pandas < 0.25.0 +flake8 +tb-paddle +tb-nightly +cma == 2.7.0 diff --git a/setup.py b/setup.py index 867d9fdf190026c4992c111e7724b0342d6169d8..93d9b1a84bee05447d077fed4bf66f81574c3ec6 100644 --- a/setup.py +++ b/setup.py @@ -36,7 +36,7 @@ REQUIRED_PACKAGES = [ ] if max_version < 3: - REQUIRED_PACKAGES += ["numpy == 1.16.0", "pandas == 0.24.0"] + REQUIRED_PACKAGES += ["numpy < 1.17.0", "pandas < 0.25.0"] else: REQUIRED_PACKAGES += ["numpy", "pandas"]