content-mlops/setup.py at develop · ipolychr/content-mlops · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
from setuptools import setup

deps = {
    'std': [
        'pydantic==1.10.8',
        'librosa==0.9.2',
        'numpy==1.24.3',
        'ffmpeg-python>=0.2.0',
        'matplotlib==3.4.3',
        'torchvision==0.12.0',
        'torch==1.11.0',
        'dvc==2.58.2',
        'smart-open==6.3.0',
        'mlflow==1.20.0',
        'optuna==3.3.0',
        'onnx==1.15.0',
        'onnxruntime==1.17.0',
        'evidently==0.4.15'
    ]
}

setup(
    name='music-genre-classification',
    version='1.0.1',
    description='Music Genre Classification',
    install_requires=[],
    extras_require=deps,
    entry_points={
        'console_scripts': [
            'pipeline = __main__.py:cli'
        ]
    },
    packages=['genre_classification']
)