setup.py

from setuptools import setup, find_packages

_REQUIRED = [
    "protobuf<4.24",
    "fsspec==2023.10.0",
    "datasets==2.15.0"
]

_OPTIONAL = {
    "train": [
        "lightning-bolts==0.7.0",
        "lightning-utilities==0.10.0"
    ],
    "dev": [
        "pytest"
    ]
}

setup(
    name='jrt',
    version="0.0.1",
    packages=find_packages(include=['based', 'based.*']),
    author="JRT",
    author_email="author@example.com",
    description="some description",
    python_requires=">=3.8",
    install_requires=_REQUIRED,
    extras_require=_OPTIONAL,
)