from setuptools import setup, find_packages

setup(
    name="neuroflex",
    version="0.1.3",  # Updated version to 0.1.3
    author="kasinadhsarma",
    author_email="kasinadhsarma@gmail.com",
    description="An advanced neural network framework with interpretability, generalization, robustness, and fairness features",
    long_description=open("README.md").read(),
    long_description_content_type="text/markdown",
    url="https://github.com/VishwamAI/neuroflex",
    packages=find_packages(exclude=["tests*"]),
    classifiers=[
        "Development Status :: 3 - Alpha",
        "Intended Audience :: Developers",
        "Intended Audience :: Science/Research",
        "License :: OSI Approved :: MIT License",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: 3.11",
        "Programming Language :: Python :: 3.12",
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
    ],
    python_requires=">=3.8",
    install_requires=[
        "absl-py",
        "adversarial-robustness-toolbox",
        "aiohappyeyeballs",
        "aiohttp",
        "aiosignal",
        "alphafold",
        "appdirs",
        "astunparse",
        "attrs",
        "autograd",
        "autoray",
        "biom-format",
        "biopython==1.80",
        "blosc2",
        "cachetools",
        "certifi",
        "charset-normalizer",
        "chex==0.1.6",
        "CHSPy",
        "click==8.0.4",
        "contextlib2",
        "deap",
        "decorator",
        "dill",
        "dm-haiku==0.0.9",
        "dm-sonnet",
        "dm-tree",
        "docker",
        "dpath",
        "ete3",
        "etils",
        "filelock",
        "flatbuffers",
        "flax==0.6.10",
        "frozenlist",
        "fsspec",
        "gast<=0.4.0,>=0.2.1",
        "google-pasta",
        "grpcio",
        "gym",
        "h5py",
        "huggingface-hub",
        "humanize",
        "idna",
        "immutabledict",
        "importlib_resources",
        "iniconfig",
        "jax==0.4.13",
        "jaxlib==0.4.13",
        "Jinja2",
        "jitcdde",
        "jitcxde_common",
        "jmp",
        "joblib",
        "keras<2.13,>=2.12.0",
        "lale==0.6.0",
        "libclang",
        "lightning-utilities",
        "llvmlite",
        "Markdown",
        "markdown-it-py",
        "MarkupSafe",
        "mdurl",
        "ml-collections",
        "ml-dtypes",
        "mpmath",
        "msgpack",
        "multidict",
        "namex",
        "natsort",
        "ndindex",
        "nest-asyncio",
        "networkx",
        "neurolib",
        "numba",
        "numexpr",
        "numpy==1.23.5",
        "nvidia-cublas-cu11",
        "nvidia-cuda-cupti-cu11",
        "nvidia-cuda-nvrtc-cu11",
        "nvidia-cuda-runtime-cu11",
        "nvidia-cudnn-cu11",
        "nvidia-cufft-cu11",
        "nvidia-curand-cu11",
        "nvidia-cusolver-cu11",
        "nvidia-cusparse-cu11",
        "nvidia-nccl-cu11",
        "nvidia-nvtx-cu11",
        "openmm==8.1.1",
        "opt-einsum",
        "optax==0.1.5",
        "optree",
        "orbax-checkpoint==0.2.1",
        "packaging",
        "pandas",
        "patsy",
        "pbr",
        "PennyLane==0.30.0",
        "PennyLane_Lightning==0.30.0",
        "pluggy",
        "protobuf==3.20.3",
        "psutil",
        "py-cpuinfo",
        "Pygments",
        "pypet",
        "pytest",
        "python-dateutil",
        "pytorch-lightning==1.9.5",
        "pytz",
        "PyYAML",
        "qiskit>=1.0.0",
        "qiskit-aer>=0.12.0",
        "qiskit-algorithms>=0.2.0",
        "regex",
        "requests",
        "rich",
        "rustworkx>=0.13.0",
        "safetensors",
        "scikit-bio>=0.5.8,<0.6.0",
        "scikit-learn==1.1.1",
        "scipy==1.9.0",
        "six",
        "statsmodels",
        "stevedore",
        "symengine",
        "sympy",
        "tables",
        "tabulate",
        "tensorboard<2.13,>=2.12",
        "tensorboard-data-server",
        "tensorflow==2.12.0",
        "tensorflow-io-gcs-filesystem",
        "tensorstore",
        "termcolor",
        "threadpoolctl",
        "tokenizers==0.13.3",
        "toml",
        "toolz",
        "torch==1.13.1",
        "torchmetrics",
        "torchvision==0.14.1",
        "tqdm",
        "transformers==4.30.2",
        "triton==2.0.0",
        "typing_extensions>=3.7.4,<4.6.0",
        "tzdata",
        "urllib3",
        "Werkzeug",
        "wrapt",
        "xarray",
        "yarl",
        "zipp",
        "PyQt5",
        "einops",
        "prophet",
        "psutil",
        "aif360",
    ],
    extras_require={
        'dev': [
            'pytest',
            'flake8',
        ],
    },
)
