diff --git a/setup.py b/setup.py index 48356c77060bbfa14a4228c75b5baa875729c765..dab03e5593659e2c7aa3410179eb1bd6b6b39a64 100644 --- a/setup.py +++ b/setup.py @@ -2,6 +2,7 @@ # -*- coding: utf-8 -*- from setuptools import setup +import sys ver_dic = {} version_file = open("pytools/version.py") @@ -12,6 +13,17 @@ finally: exec(compile(version_file_contents, "pytools/version.py", "exec"), ver_dic) +version = sys.version_info +requirements = [ + "decorator>=3.2.0", + "appdirs>=1.4.0", + "six>=1.8.0", + "numpy>=1.6.0", + ] + +if version[0] == 3 and version[1] = 6: + requirements.append("dataclasses") + setup(name="pytools", version=ver_dic["VERSION_TEXT"], description="A collection of tools for Python", @@ -37,12 +49,7 @@ setup(name="pytools", python_requires="~=3.6", - install_requires=[ - "decorator>=3.2.0", - "appdirs>=1.4.0", - "six>=1.8.0", - "numpy>=1.6.0", - ], + install_requires=requirements, package_data={"pytools": ["py.typed"]},