SPECTER/setup.py at main · rikab/SPECTER · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
from setuptools import setup, find_packages
import os
import subprocess

# Function to dynamically fetch the version from VCS (git in this case)
def get_version():
    try:
        # Get the output from git describe
        version = subprocess.check_output(["git", "describe", "--tags", "--long"]).strip().decode("utf-8")

        # Match the version and discard the local part
        match = re.match(r"v?(\d+\.\d+\.\d+)(?:-(\d+)-g([0-9a-f]+))?", version)
        if match:
            base_version = match.group(1)  # 0.0.1
            # If there are additional commits, append post-release versioning
            if match.group(2) and match.group(2) != "0":  # Not exactly at the tag
                version = f"{base_version}.post{match.group(2)}"  # 0.0.1.post3
            else:
                version = base_version  # If it's exactly a tagged version, use it directly
        return version
    except Exception:
        return "0.0.1"  # Default/fallback version if VCS version not available


setup(
    name="specterpy",
    version=get_version(),  # Dynamic versioning based on VCS
    description="Framework for defining, building, and evaluating generalized shape observables for collider physics",
    long_description=open("README.md").read(),
    long_description_content_type="text/markdown",
    author="Rikab Gambhir",
    author_email="rikab@mit.edu",
    url="https://github.com/rikab/SPECTER",
    project_urls={
        "Documentation": "https://github.com/rikab/SPECTER",
        "Homepage": "https://github.com/rikab/SPECTER",
        "Issue Tracker": "https://github.com/rikab/SPECTER/issues",
        "Releases": "https://github.com/rikab/SPECTER/releases",
        "Source Code": "https://github.com/rikab/SPECTER",
    },
    packages=find_packages(where="src"),  # Assume the package is in the "src" directory
    package_dir={"": "src"},  # Maps the root package to src/
    python_requires=">=3.7",
    install_requires=[
        "jax>=0.4.13",
        "jaxlib>=0.4.13",
        "scipy>=1.5.1",
        "matplotlib>=3.5.0",
        "numpy",  # scipy controls compatible numpy versions
        "rikabplotlib",
        "particleloader"
    ],
    classifiers=[
        "Development Status :: 3 - Alpha",
        "Intended Audience :: Science/Research",
        "License :: OSI Approved :: MIT License",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3 :: Only",
        "Programming Language :: Python :: 3.7",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: Implementation :: CPython",
        "Topic :: Scientific/Engineering",
        "Topic :: Scientific/Engineering :: Physics",
    ],
    keywords=["shape observables", "jet physics"],
    license="MIT",
    include_package_data=True,
    # Optional build hook equivalent, though not strictly necessary for this simple setup.py
    data_files=[
        ("", ["LICENSE", "README.md", "pyproject.toml"])
    ],
    zip_safe=False,
)