initial commit

setup.py

@@ -0,0 +1,88 @@
+# 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.7"
+
+# We use typeguard internally (in a fairly minimal way), but it's not required that
+# end users make the same choice.
+install_requires = ["jax>=0.3.4", "typeguard>=2.13.3"]
+
+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],
+)