thoutam's picture
Add setup.py for pip package and pipeline
7b9813f verified
#!/usr/bin/env python3
"""
Setup script for StoxChai NSE Stock Prediction Models
"""
from setuptools import setup, find_packages
import os
# Read the README file
def read_readme():
with open("README.md", "r", encoding="utf-8") as fh:
return fh.read()
# Read requirements
def read_requirements():
with open("requirements.txt", "r", encoding="utf-8") as fh:
return [line.strip() for line in fh if line.strip() and not line.startswith("#")]
setup(
name="stoxchai-nse-predictor",
version="1.0.0",
author="StoxChai Team",
author_email="team@stoxchai.com",
description="NSE Stock Price Prediction Models using Machine Learning",
long_description=read_readme(),
long_description_content_type="text/markdown",
url="https://huggingface.co/thoutam/stoxchai-nse-predictor",
project_urls={
"Bug Reports": "https://github.com/stoxchai/nse-predictor/issues",
"Source": "https://github.com/stoxchai/nse-predictor",
"Documentation": "https://huggingface.co/thoutam/stoxchai-nse-predictor",
},
packages=find_packages(),
classifiers=[
"Development Status :: 5 - Production/Stable",
"Intended Audience :: Financial and Insurance Industry",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Topic :: Office/Business :: Financial :: Investment",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Scientific/Engineering :: Information Analysis",
],
python_requires=">=3.8",
install_requires=read_requirements(),
extras_require={
"dev": [
"pytest>=6.0",
"pytest-cov>=2.0",
"black>=21.0",
"flake8>=3.8",
"mypy>=0.800",
],
"docs": [
"sphinx>=4.0",
"sphinx-rtd-theme>=0.5",
"myst-parser>=0.15",
],
"full": [
"torch>=2.0.0",
"tensorflow>=2.13.0",
],
},
include_package_data=True,
package_data={
"stoxchai_nse_predictor": [
"models/*.joblib",
"data/*.json",
"*.md",
"*.txt",
],
},
entry_points={
"console_scripts": [
"stoxchai-predict=stoxchai_nse_predictor.cli:main",
],
},
keywords=[
"stock-prediction",
"machine-learning",
"nse",
"india",
"finance",
"trading",
"ai",
"ml",
"regression",
"time-series",
],
zip_safe=False,
)