setup.py

# Copyright 2023, Yahoo Inc.
# Licensed under the Apache License, Version 2.0.
# See the accompanying LICENSE file for terms.

import os
import glob
from setuptools import setup, find_namespace_packages

SCRIPTS_DIR = "bin"
PACKAGE_DIR = "spivak"
PYTHON_REQUIRES = ">=3.6"
INSTALL_REQUIRES = [
    "numpy>=1.18.5,<1.23.0",
    "scipy>=1.4.1,<1.8.0",
    "scikit-learn>=0.24.2,<1.1.0",
    "pandas>=1.1.5,<1.4.0",
    "Pillow>=8.4.0,<9.1.0",
    "opencv-python>=4.5.4.58,<4.8.0",
    "tqdm~=4.62.3",
    # Note: some portion of the code here does not depend on TensorFlow,
    # so if you are not interested in using the models, you probably
    # don't really have to install TensorFlow.
    "tensorflow~=2.7.0",
    "tensorboard~=2.7.0",
    # This version of tensorflow_probability is needed so that it works with
    # Tensorflow 2.3. It also happens to work with 2.7. If you are using 2.7
    # or more recent versions of TensorFlow, you can probably upgrade this.
    "tensorflow_probability~=0.11.1",
    # tensorflow-addons has the decoupled weight decay functionality.
    # For TensorFlow 2.3, we need to use this older version (0.13) of the
    # addons package. This version also works with TensorFlow 2.7, even though
    # it will print out some warnings.
    "tensorflow-addons~=0.13.0",
    "typing_extensions==4.7.1",
    "plotly>=5.4.0,<5.6.0",
    # kaleido is used for being able to render pdf plots with plotly.
    "kaleido~=0.2.1",
    # scikit-video (skvideo), moviepy, imutils are only used in
    # spivak.feature_extraction.SoccerNetDataLoader.py.
    "scikit-video~=1.1.11",
    "moviepy~=1.0.3",
    "imageio-ffmpeg==0.4.9",
    "imutils~=0.5.4",
    # packaging is only used by
    # spivak.models.assembly.huggingface_activations.py.
    "packaging>=21.2,<21.4"
]
EXTRAS_REQUIRES = {
    "av": [
        # av is used only in bin/create_visualizations.py and inside
        # spivak.video_visualization, for creating videos with visualizations
        # of results. It requires the ffmpeg libraries, so we leave it as an
        # optional extra.
        "av~=10.0.0"
    ]
}
packages = find_namespace_packages(include=[f"{PACKAGE_DIR}.*"])
scripts = glob.glob(os.path.join(SCRIPTS_DIR, "*.py"))
setup(
    name=PACKAGE_DIR,
    packages=packages,
    python_requires=PYTHON_REQUIRES,
    install_requires=INSTALL_REQUIRES,
    extras_require=EXTRAS_REQUIRES,
    scripts=scripts,
)