45e1175b78
* Exclude benchmarks directory from distribution * lint * format with bash shell/format.sh
68 lines
1.8 KiB
Python
68 lines
1.8 KiB
Python
"""Setup script."""
|
|
|
|
import os
|
|
import pathlib
|
|
|
|
from setuptools import find_packages
|
|
from setuptools import setup
|
|
|
|
|
|
def read(rel_path):
|
|
here = os.path.abspath(os.path.dirname(__file__))
|
|
with open(os.path.join(here, rel_path)) as fp:
|
|
return fp.read()
|
|
|
|
|
|
def get_version(rel_path):
|
|
for line in read(rel_path).splitlines():
|
|
if line.startswith("__version__"):
|
|
delim = '"' if '"' in line else "'"
|
|
return line.split(delim)[1]
|
|
raise RuntimeError("Unable to find version string.")
|
|
|
|
|
|
HERE = pathlib.Path(__file__).parent
|
|
README = (HERE / "README.md").read_text()
|
|
VERSION = get_version("keras/src/version.py")
|
|
|
|
setup(
|
|
name="keras",
|
|
description="Multi-backend Keras.",
|
|
long_description_content_type="text/markdown",
|
|
long_description=README,
|
|
version=VERSION,
|
|
url="https://github.com/keras-team/keras",
|
|
author="Keras team",
|
|
author_email="keras-users@googlegroups.com",
|
|
license="Apache License 2.0",
|
|
install_requires=[
|
|
"absl-py",
|
|
"numpy",
|
|
"rich",
|
|
"namex",
|
|
"h5py",
|
|
"optree",
|
|
"ml-dtypes",
|
|
"packaging",
|
|
],
|
|
# Supported Python versions
|
|
python_requires=">=3.9",
|
|
classifiers=[
|
|
"Development Status :: 4 - Beta",
|
|
"Programming Language :: Python :: 3",
|
|
"Programming Language :: Python :: 3.9",
|
|
"Programming Language :: Python :: 3.10",
|
|
"Programming Language :: Python :: 3.11",
|
|
"Programming Language :: Python :: 3 :: Only",
|
|
"Operating System :: Unix",
|
|
"Operating System :: MacOS",
|
|
"Intended Audience :: Science/Research",
|
|
"Topic :: Scientific/Engineering",
|
|
"Topic :: Software Development",
|
|
],
|
|
packages=find_packages(
|
|
include=("keras", "keras.*"),
|
|
exclude=("*_test.py", "benchmarks"),
|
|
),
|
|
)
|