import setuptools

with open("README.md", "r") as fh:
    long_description = fh.read()

setuptools.setup(
    name="cdtools",
    version="0.3.0",
    python_requires='>3.8', # recommended minimum version for pytorch 2.3.0
    author="Abe Levitan",
    author_email="abraham.levitan@psi.ch",
    description="Tools for coherent diffractive imaging and ptychography",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.mit.edu/scattering/CDTools.git",
    install_requires=[
        "numpy>=1.0",
        "scipy>=1.0",
        "matplotlib>=2.0", # 2.0 has better colormaps which are used by default
        "python-dateutil",
        "torch>=2.3.0", #2.3.0 is the earliest release for which L-BFGS works directly on complex-valued leaf tensors
        "h5py>=2.1"],
    extras_require={
        'tests': [
            "pytest",
            "pooch",
        ],
        'docs': [
            "sphinx>=4.3.0",
            "sphinx-argparse",
            "sphinx_rtd_theme>=0.5.1"
        ]
    },
    package_dir={"": "src"},
    packages=setuptools.find_packages("src"),
    classifiers=[
        "Programming Language :: Python :: 3",
        "Operating System :: OS Independent",
    ],
)