setup.py

import shlex
from subprocess import check_call

from setuptools import find_packages
from setuptools import setup
from setuptools.command.develop import develop

# Create post develop command class for hooking into the python setup process
# This command will run after dependencies are installed


class PostDevelopCommand(develop):
    def run(self):
        try:
            check_call(shlex.split('pre-commit install'))
        except Exception as e:
            print(f"Unable to run 'pre-commit install' with exception {e}")
        develop.run(self)


install_requires = [
    'numpy',
    'geopandas',
    'shapely',
    'rasterio',
    'pyproj',
    'pygc',
    'aiohttp',
    'wandb',
    'zarr',
    'gcsfs',
    'torch @ https://download.pytorch.org/whl/cu110/torch-1.7.1%2Bcu110-cp38-cp38-linux_x86_64.whl',
    'torchvision @ https://download.pytorch.org/whl/cu110/torchvision-0.8.2%2Bcu110-cp38-cp38-linux_x86_64.whl',
    'google-cloud-bigquery',
    'rioxarray',
    'progressbar2',
    'pyarrow',
    'google-cloud-pubsub',
    'kornia',
    'loguru',
    'torchmetrics',
    'dask',
    'distributed',
]  # alternatively, read from `requirements.txt`

with open('src/wfe/seco/requirements.txt') as fp:
    seco_requires = fp.readlines()

print(seco_requires)

extra_requires = ['earthengine-api']  # optional dependencies
test_requires = ['pytest']  # test dependencies
dev_requires = ['pre-commit']

setup(
    name='wfe',
    version='0.1.0',
    package_dir={'': 'src'},
    packages=find_packages(where='src'),
    install_requires=install_requires,
    extras_require={
        'test': test_requires,
        'extra': extra_requires,
        'seco': seco_requires,
        'dev': test_requires + dev_requires + seco_requires,
    },
    cmdclass={'develop': PostDevelopCommand},
    include_package_data=True,
    package_data={'': ['wfe/data/s2extract/resources/*']},

)