parcus/setup.py at main · theokoles7/parcus · 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
76
77
78
79
80
81
82
83
84
"""# parcus.setup

Package setup utility.
"""

from pathlib    import Path
from setuptools import find_packages, setup
from typing     import Any, Dict


# HELPERS ==========================================================================================

def get_long_description() -> str:
    """# Get Long Description.

    ## Returns:
        * str:  README file contents.
    """
    with open(Path(__file__).parent / "README.md", encoding = "utf-8") as f: return f.read()


def get_version() -> str:
    """# Get Package Version.

    ## Returns:
        * str:  Current package version.
    """
    # Initialize metadata mapping.
    metadata:   Dict[str, Any] =    {}

    # Open metadata file.
    with open(Path(__file__).parent / "parcus" / "__meta__.py") as f:

        # Map package data.
        exec(f.read(), metadata)

    # Provide package version.
    return metadata["__version__"]


# SETUP UTILITY ====================================================================================

setup(
    name =                          "parcus",
    version =                       get_version(),
    author =                        "Gabriel C. Trahan",
    author_email =                  "gabriel.trahan1@louisiana.edu",
    description =                   """Experiments in analyzing the correlation and effects of token
                                    budgets on a language model's ability to reason and generate
                                    accurate responses.""",
    long_description =               get_long_description(),
    long_description_content_type = "text/markdown",
    license =                       "GNU GENERAL PUBLIC LICENSE Version 3, 29 June 2007",
    license_files =                 ("LICENSE"),
    url =                           "https://github.com/theokoles7/parcus",
    packages =                      find_packages(),
    python_requires =               ">=3.10",
    install_requires =              [
                                        "accelerate",
                                        "bitsandbytes",
                                        "datasets",
                                        "matplotlib",
                                        "numpy",
                                        "torch",
                                        "tqdm",
                                        "transformers",
                                    ],
    entry_points =                  {
                                        "console_scripts":  [
                                                                "parcus=parcus.__main__:parcus_entry_point"
                                                            ],
                                    },
    classifiers =                   [
                                        "Topic :: Scientific/Engineering :: Artificial Intelligence",
                                        "License :: OSI Approved :: GNU General Public License v3 (GPLv3)",
                                        "Intended Audience :: Developers",
                                        "Operating System :: OS Independent",
                                        "Programming Language :: Python :: 3",
                                        "Programming Language :: Python :: 3.10",
                                        "Programming Language :: Python :: 3.11",
                                        "Programming Language :: Python :: 3.12",
                                        "Programming Language :: Python :: 3.13",
                                    ]
)