| from setuptools import setup, find_packages |
|
|
| setup( |
| name="self-healing-training", |
| version="1.0.0", |
| description="Fully autonomous self-healing layer for Hugging Face TRL trainers", |
| long_description=open("README.md").read(), |
| long_description_content_type="text/markdown", |
| author="Autonomous ML Intern", |
| url="https://huggingface.co/ScottzillaSystems/self-healing-training", |
| packages=find_packages(), |
| python_requires=">=3.9", |
| install_requires=[ |
| "torch>=2.0.0", |
| "transformers>=4.40.0", |
| "trl>=0.9.0", |
| ], |
| extras_require={ |
| "trackio": ["trackio"], |
| "dev": ["pytest", "pytest-cov", "black", "ruff"], |
| }, |
| classifiers=[ |
| "Development Status :: 4 - Beta", |
| "Intended Audience :: Science/Research", |
| "License :: OSI Approved :: MIT License", |
| "Programming Language :: Python :: 3", |
| "Programming Language :: Python :: 3.9", |
| "Programming Language :: Python :: 3.10", |
| "Programming Language :: Python :: 3.11", |
| "Programming Language :: Python :: 3.12", |
| "Topic :: Scientific/Engineering :: Artificial Intelligence", |
| ], |
| ) |
