setup.py

from setuptools import find_packages, setup

with open("VERSION") as version_file:
    version = version_file.read().strip()

extras_require_serving = [
    "lightgbm~=3.3.2",
    "xgboost~=0.90",
    "treelite_runtime~=2.2.2",
    "scikit-learn~=1.2.0",
    "cloudpickle~=2.1.0",
    "opentelemetry-instrumentation-grpc~=0.38b0",
    "opentelemetry-api>=1.17.0, <2.0",
    "opentelemetry-sdk>=1.17.0, <2.0",
    "opentelemetry-exporter-otlp>=1.17.0, <2.0",
]
extras_require_test = [
    *extras_require_serving,
    "pytest~=6.2",
    "black~=22.6.0",
    "isort~=5.10.1",
    "flake8~=4.0.1",
    "flake8-mutable~=1.2.0",
    "mypy==0.991",
    "mypy-protobuf~=3.4.0",
    "types-protobuf",
    "pyupgrade~=3.3.1",
]
extras_require_dev = [
    *extras_require_test,
    "treelite~=2.4",
    "pre-commit~=2.20.0",
    "pip-tools~=6.12.3",
]

setup(
    name="pythie-serving",
    version=version,
    description="A GRPC server to serve model types using tensorflow-serving .proto services",
    author="wiremind data science team",
    author_email="data-science@wiremind.io",
    url="https://gitlab.cayzn.com/wiremind/data-science/pythie-neos.git",
    packages=find_packages(where="src"),
    package_dir={"": "src"},
    include_package_data=True,
    zip_safe=True,
    python_requires=">=3.11",
    install_requires=[
        "numpy~=1.23.0",
        "grpcio~=1.51",
        "protobuf~=4.22",
    ],
    # pip-compile setup.py --no-emit-index-url --upgrade --rebuild
    # pip-compile setup.py --no-emit-index-url --upgrade --extra serving -o pythie-serving-requirements.txt
    extras_require={
        "serving": extras_require_serving,
        "test": extras_require_test,
        "dev": extras_require_dev,
        "all": extras_require_serving + extras_require_dev,
    },
    entry_points={
        "console_scripts": [
            "pythie-serving=pythie_serving.run:run",
        ]
    },
    classifiers=[
        "Programming Language :: Python :: 3.11",
        "Programming Language :: Python :: 3 :: Only",
        "License :: OSI Approved :: GNU Lesser General Public License v3 or later (LGPLv3+)",
    ],
)