from setuptools import setup, find_packages

setup(
    name="lumen-irmas",
    version="0.1.0",
    description="LUMEN Data Science nagradni zadatak",
    author="Karlo Pintaric i Tatjana Cigula",
    packages=find_packages(include=["src"]),
    python_requires=">=3.9",
    install_requires=[
        "numpy==1.23.5",
        "transformers==4.27.4",
    ],
    extras_require={
        "backend": ["fastapi==0.95.1", "uvicorn==0.21.1", "pydantic==1.10.7", "python-multipart==0.0.6"],
        "frontend": ["streamlit==1.21.0", "requests==2.28.2", "soundfile==0.12.1"],
        "user": [
            "lumen-irmas[backend]",
            "lumen-irmas[frontend]",
            "torch==1.13.1",
            "torchaudio==0.13.1",
            "torchvision==0.14.1",
            ],
        "dev": [
            "lumen-irmas[user]",
            "librosa==0.10.0.post2",
            "pandas==1.5.3",
            "scikit-learn==1.2.2",
            "tqdm==4.65.0",
            "wandb==0.14.2",
            "pytest==7.3.1",
            "joblib==1.2.0",
            "PyYAML==6.0",
            "flake8==6.0.0",
            "isort== 5.12.0",
            "black==23.3.0"
            ]
    },
)