setup.py

from setuptools import setup, find_packages
import tnlearn

with open("README.md", "r", encoding="utf-8") as fh:
    long_description = fh.read()

VERSION = tnlearn.__version__

setup(
    name="tnlearn",
    version=VERSION,
    author="Meng WANG",
    author_email="wangmeng22@stu.hit.edu.cn",
    description="A Python package that uses task-based neurons to build neural networks.",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/NewT123-WM/tnlearn",  
    packages=find_packages(),
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: BSD License",
        "Operating System :: OS Independent",
    ],
    python_requires='>=3.8',
    install_requires=[
        'torch>=2.1.0',
        'numpy>=1.26.4',
        'scikit-learn>=1.4.1.post1',
        'pandas>=2.2.1',
        'sympy>=1.12',
        'tqdm>=4.66.2',
        'matplotlib>=3.8.3',
        'ipython>=8.18.1',
        'torchinfo>=1.8.0',
        'h5py>=3.10.0',
        'setuptools>=68.2.2'
    ],
)