setup.py

from setuptools import find_packages, setup

# Base requirements for the custom transformers model
base_requirements = [
    "bytelatent @ git+https://github.com/facebookresearch/blt",
    # "bytelatent @ git+https://github.com/Vectorrent/blt.git@fix-realtime-entropy-patching",
    # "bytelatent @ file:///home/crow/repos/blt",
    "hivemind @ git+https://github.com/learning-at-home/hivemind.git@213bff98a62accb91f254e2afdccbf1d69ebdea9",
    "torch",
    "transformers",
]

# Additional requirements for training and orchestration
host_requirements = [
    "asciichartpy",
    "blessed",
    "datasets",
    "evaluate",
    "flask",
    "lightning",
    "lm_eval",
    "pytorch_optimizer",
    "werkzeug",
]

# Requirements primarily for IDE-related tooling
dev_requirements = [
    "attrs",
    "isort",
    "lsprotocol",
    "matplotlib",
    "networkx",
    "pytest",
    "pygls",
    "wandb",
]

setup(
    name="praxis",
    version="0.1.0",
    decription="as above, so below",
    author="Ryan Brooks",
    author_email="ryan@src.eco",
    url="https://src.eco",
    packages=find_packages(),
    license="MIT",
    install_requires=base_requirements,
    extras_require={
        "all": base_requirements + host_requirements + dev_requirements,
        "dev": dev_requirements,
    },
)