mirror of
https://github.com/wassname/jaxtyping.git
synced 2026-08-21 11:16:29 +08:00
101 lines
3.1 KiB
Python
101 lines
3.1 KiB
Python
# Copyright (c) 2022 Google LLC
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a copy of
|
|
# this software and associated documentation files (the "Software"), to deal in
|
|
# the Software without restriction, including without limitation the rights to
|
|
# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
|
|
# the Software, and to permit persons to whom the Software is furnished to do so,
|
|
# subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included in all
|
|
# copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
|
# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
|
|
# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
|
|
# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
|
# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
|
|
import pathlib
|
|
import re
|
|
|
|
import setuptools
|
|
|
|
|
|
_here = pathlib.Path(__file__).resolve().parent
|
|
|
|
|
|
name = "jaxtyping"
|
|
|
|
# for simplicity we actually store the version in the __version__ attribute in the
|
|
# source
|
|
with open(_here / name / "__init__.py") as f:
|
|
meta_match = re.search(r"^__version__ = ['\"]([^'\"]*)['\"]", f.read(), re.M)
|
|
if meta_match:
|
|
version = meta_match.group(1)
|
|
else:
|
|
raise RuntimeError("Unable to find __version__ string.")
|
|
|
|
author = "Patrick Kidger"
|
|
|
|
author_email = "contact@kidger.site"
|
|
|
|
description = (
|
|
"Type annotations and runtime checking for shape and dtype of JAX "
|
|
"arrays, and PyTrees."
|
|
)
|
|
|
|
with open(_here / "README.md", "r") as f:
|
|
readme = f.read()
|
|
|
|
url = "https://github.com/google/jaxtyping"
|
|
|
|
license = "MIT"
|
|
|
|
classifiers = [
|
|
"Development Status :: 3 - Alpha",
|
|
"Intended Audience :: Science/Research",
|
|
"License :: OSI Approved :: MIT License",
|
|
"Natural Language :: English",
|
|
"Programming Language :: Python :: 3",
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
|
"Topic :: Scientific/Engineering :: Mathematics",
|
|
]
|
|
|
|
python_requires = "~=3.8"
|
|
|
|
# We use typeguard internally (in a fairly minimal way), but it's not required that
|
|
# end users make the same choice.
|
|
# For typing_extensions, we choose versions that match
|
|
# https://github.com/explosion/confection/blob/main/setup.cfg#L33 used in colab
|
|
|
|
install_requires = [
|
|
"numpy>=1.20.0",
|
|
"typeguard>=2.13.3",
|
|
"typing_extensions>=3.7.4.1",
|
|
]
|
|
|
|
entry_points = dict(pytest11=["jaxtyping = jaxtyping.pytest_plugin"])
|
|
|
|
setuptools.setup(
|
|
name=name,
|
|
version=version,
|
|
author=author,
|
|
author_email=author_email,
|
|
maintainer=author,
|
|
maintainer_email=author_email,
|
|
description=description,
|
|
long_description=readme,
|
|
long_description_content_type="text/markdown",
|
|
url=url,
|
|
license=license,
|
|
classifiers=classifiers,
|
|
zip_safe=False,
|
|
python_requires=python_requires,
|
|
install_requires=install_requires,
|
|
entry_points=entry_points,
|
|
packages=[name],
|
|
include_package_data=True,
|
|
)
|