1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
|
import setuptools
with open("README.md", "r", encoding="utf-8") as fh:
long_description = fh.read()
setuptools.setup(
name="opt_einsum_fx",
version="0.1.4", # remember to update in `__init__.py` too!
author="Linux-cpp-lisp",
url="https://github.com/Linux-cpp-lisp/opt_einsum_fx",
description="Einsum optimization using opt_einsum and PyTorch FX",
long_description=long_description,
long_description_content_type="text/markdown",
license="MIT",
license_files="LICENSE",
project_urls={
"Bug Tracker": "https://github.com/Linux-cpp-lisp/opt_einsum_fx/issues",
},
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
"Development Status :: 4 - Beta",
],
python_requires=">=3.6",
install_requires=["torch>=1.8.0", "opt_einsum", "packaging"],
packages=["opt_einsum_fx"],
)
|