setup.py

from setuptools import setup, find_packages
import os

# Read README for long description
def read_readme():
    readme_path = os.path.join(os.path.dirname(__file__), 'README.md')
    if os.path.exists(readme_path):
        with open(readme_path, 'r', encoding='utf-8') as f:
            return f.read()
    return ''

# Read requirements
def read_requirements():
    req_path = os.path.join(os.path.dirname(__file__), 'requirements.txt')
    requirements = []
    if os.path.exists(req_path):
        with open(req_path, 'r', encoding='utf-8') as f:
            for line in f:
                line = line.strip()
                if line and not line.startswith('#'):
                    requirements.append(line)
    return requirements

setup(
    name='ctda-cpa',
    version='1.0.0',
    author='Tong Yang, Ning Li, Bo Huang, Jianmei Guo',
    author_email='tyang2001@163.com',
    description='Cross-Thread Dependency-Aware Critical Path Analysis for Large Model Training',
    long_description=read_readme(),
    long_description_content_type='text/markdown',
    url='https://github.com/yourusername/CTDA-CPA',
    packages=find_packages(exclude=['tests', 'docs', 'examples', 'optimizations']),
    classifiers=[
        'Development Status :: 4 - Beta',
        'Intended Audience :: Developers',
        'Intended Audience :: Science/Research',
        'Topic :: Scientific/Engineering :: Artificial Intelligence',
        'License :: OSI Approved :: MIT License',
        'Programming Language :: Python :: 3',
        'Programming Language :: Python :: 3.8',
        'Programming Language :: Python :: 3.9',
        'Programming Language :: Python :: 3.10',
        'Programming Language :: Python :: 3.11',
    ],
    python_requires='>=3.8',
    install_requires=read_requirements(),
    extras_require={
        'distributed': [
            'deepspeed==0.14.4',
            'transformers==4.41.1',
            'datasets==2.14.6',
            'accelerate==0.30.1',
        ],
    },
    entry_points={
        'console_scripts': [
            'ctda-cpa=ctda_cpa.hotspot_analysis:main',
        ],
    },
    include_package_data=True,
    zip_safe=False,
)