Abeshith's picture
Add pipeline stages implementation
a7d80f2
from setuptools import setup, find_packages
setup(
name="mlpipeline",
version="0.1.0",
author="AutoML Team",
packages=find_packages(where="src"),
package_dir={"": "src"},
python_requires=">=3.11",
install_requires=[
"fastapi",
"uvicorn[standard]",
"pydantic",
"pandas",
"numpy",
"scikit-learn",
"autogluon.tabular",
"flaml",
"pycaret",
"mlflow",
"dvc",
"evidently",
"pyyaml",
"python-box",
"ensure",
"kaggle",
],
)