array-api-jit 1.0.0__tar.gz → 1.1.0__tar.gz

{array_api_jit-1.0.0/src/array_api_jit.egg-info → array_api_jit-1.1.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: array-api-jit
-Version: 1.0.0
+Version: 1.1.0
 Summary: JIT decorator supporting multiple array API compatible libraries
-Author-email: 34j <34j.95a2p@simplelogin.com>
-License: MIT
+Author: 34j
+License-Expression: MIT
 Project-URL: Bug Tracker, https://github.com/34j/array-api-jit/issues
 Project-URL: Changelog, https://github.com/34j/array-api-jit/blob/main/CHANGELOG.md
 Project-URL: documentation, https://array-api-jit.readthedocs.io
@@ -12,12 +12,12 @@ Classifier: Development Status :: 2 - Pre-Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Natural Language :: English
 Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Software Development :: Libraries
-Requires-Python: >=3.10
+Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: array-api-compat>=1.11.2
@@ -32,19 +32,19 @@ Dynamic: license-file
   <a href="https://array-api-jit.readthedocs.io">
     <img src="https://img.shields.io/readthedocs/array-api-jit.svg?logo=read-the-docs&logoColor=fff&style=flat-square" alt="Documentation Status">
   </a>
+</p>
+<p align="center">
   <a href="https://codecov.io/gh/34j/array-api-jit">
     <img src="https://img.shields.io/codecov/c/github/34j/array-api-jit.svg?logo=codecov&logoColor=fff&style=flat-square" alt="Test coverage percentage">
   </a>
-</p>
-<p align="center">
   <a href="https://github.com/astral-sh/uv">
     <img src="https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/uv/main/assets/badge/v0.json" alt="uv">
   </a>
   <a href="https://github.com/astral-sh/ruff">
     <img src="https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json" alt="Ruff">
   </a>
-  <a href="https://github.com/pre-commit/pre-commit">
-    <img src="https://img.shields.io/badge/pre--commit-enabled-brightgreen?logo=pre-commit&logoColor=white&style=flat-square" alt="pre-commit">
+  <a href="https://github.com/j178/prek">
+    <img src="https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/j178/prek/master/docs/assets/badge-v0.json" alt="prek">
   </a>
 </p>
 <p align="center">
@@ -78,6 +78,8 @@ pip install array-api-jit
 Simply decorate your function with `@jit()`:
 
 ```python
+from typing import Any
+
 from array_api_jit import jit
 
 
@@ -112,25 +114,3 @@ def sin_n_times(x: Any, n: int) -> Any:
         x = xp.sin(x)
     return x
 ```
-
-## Contributors ✨
-
-Thanks goes to these wonderful people ([emoji key](https://allcontributors.org/docs/en/emoji-key)):
-
-<!-- prettier-ignore-start -->
-<!-- ALL-CONTRIBUTORS-LIST:START - Do not remove or modify this section -->
-<!-- markdownlint-disable -->
-<!-- markdownlint-enable -->
-<!-- ALL-CONTRIBUTORS-LIST:END -->
-<!-- prettier-ignore-end -->
-
-This project follows the [all-contributors](https://github.com/all-contributors/all-contributors) specification. Contributions of any kind welcome!
-
-## Credits
-
-[![Copier](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/copier-org/copier/master/img/badge/badge-grayscale-inverted-border-orange.json)](https://github.com/copier-org/copier)
-
-This package was created with
-[Copier](https://copier.readthedocs.io/) and the
-[browniebroke/pypackage-template](https://github.com/browniebroke/pypackage-template)
-project template.

{array_api_jit-1.0.0 → array_api_jit-1.1.0}/README.md

@@ -7,19 +7,19 @@
   <a href="https://array-api-jit.readthedocs.io">
     <img src="https://img.shields.io/readthedocs/array-api-jit.svg?logo=read-the-docs&logoColor=fff&style=flat-square" alt="Documentation Status">
   </a>
+</p>
+<p align="center">
   <a href="https://codecov.io/gh/34j/array-api-jit">
     <img src="https://img.shields.io/codecov/c/github/34j/array-api-jit.svg?logo=codecov&logoColor=fff&style=flat-square" alt="Test coverage percentage">
   </a>
-</p>
-<p align="center">
   <a href="https://github.com/astral-sh/uv">
     <img src="https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/uv/main/assets/badge/v0.json" alt="uv">
   </a>
   <a href="https://github.com/astral-sh/ruff">
     <img src="https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json" alt="Ruff">
   </a>
-  <a href="https://github.com/pre-commit/pre-commit">
-    <img src="https://img.shields.io/badge/pre--commit-enabled-brightgreen?logo=pre-commit&logoColor=white&style=flat-square" alt="pre-commit">
+  <a href="https://github.com/j178/prek">
+    <img src="https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/j178/prek/master/docs/assets/badge-v0.json" alt="prek">
   </a>
 </p>
 <p align="center">
@@ -53,6 +53,8 @@ pip install array-api-jit
 Simply decorate your function with `@jit()`:
 
 ```python
+from typing import Any
+
 from array_api_jit import jit
 
 
@@ -87,25 +89,3 @@ def sin_n_times(x: Any, n: int) -> Any:
         x = xp.sin(x)
     return x
 ```
-
-## Contributors ✨
-
-Thanks goes to these wonderful people ([emoji key](https://allcontributors.org/docs/en/emoji-key)):
-
-<!-- prettier-ignore-start -->
-<!-- ALL-CONTRIBUTORS-LIST:START - Do not remove or modify this section -->
-<!-- markdownlint-disable -->
-<!-- markdownlint-enable -->
-<!-- ALL-CONTRIBUTORS-LIST:END -->
-<!-- prettier-ignore-end -->
-
-This project follows the [all-contributors](https://github.com/all-contributors/all-contributors) specification. Contributions of any kind welcome!
-
-## Credits
-
-[![Copier](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/copier-org/copier/master/img/badge/badge-grayscale-inverted-border-orange.json)](https://github.com/copier-org/copier)
-
-This package was created with
-[Copier](https://copier.readthedocs.io/) and the
-[browniebroke/pypackage-template](https://github.com/browniebroke/pypackage-template)
-project template.

array_api_jit-1.1.0/pyproject.toml

@@ -0,0 +1,62 @@
+[build-system]
+build-backend = "setuptools.build_meta"
+requires = [ "setuptools>=77.0.3" ]
+
+[project]
+name = "array-api-jit"
+version = "1.1.0"
+description = "JIT decorator supporting multiple array API compatible libraries"
+readme = "README.md"
+license = "MIT"
+authors = [
+  { name = "34j" },
+]
+requires-python = ">=3.12"
+classifiers = [
+  "Development Status :: 2 - Pre-Alpha",
+  "Intended Audience :: Developers",
+  "Natural Language :: English",
+  "Operating System :: OS Independent",
+  "Programming Language :: Python :: 3 :: Only",
+  "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
+  "Programming Language :: Python :: 3.14",
+  "Topic :: Software Development :: Libraries",
+]
+dependencies = [
+  "array-api-compat>=1.11.2",
+]
+urls."Bug Tracker" = "https://github.com/34j/array-api-jit/issues"
+urls.Changelog = "https://github.com/34j/array-api-jit/blob/main/CHANGELOG.md"
+urls.documentation = "https://array-api-jit.readthedocs.io"
+urls.repository = "https://github.com/34j/array-api-jit"
+
+[dependency-groups]
+dev = [
+  "cm-time>=0.1.2",
+  "jax>=0.4.30",
+  "numba>=0.60",
+  "packaging-34j[test]>=1.1.8",
+  "torch>=2.7.1",
+  "ty>=0.0.99",
+]
+docs = [
+  "packaging-34j[docs]",
+]
+
+[tool.ruff]
+extend = ".venv/ruff.toml"
+lint.extend-per-file-ignores."/**/src/**/*" = [
+  "C901",
+  "PLC0415",
+]
+lint.extend-per-file-ignores."/**/tests/**/*" = [
+  "PLC0415",
+  "RUF005",
+]
+
+[tool.pyproject-fmt]
+max_supported_python = "3.14"
+
+[tool.ty]
+terminal.error-on-warning = true

{array_api_jit-1.0.0 → array_api_jit-1.1.0}/src/array_api_jit/_main.py

@@ -19,8 +19,8 @@ if importlib.util.find_spec("numba"):
     from numba.extending import overload
 
     @overload(array_namespace)
-    def _array_namespace_overload(*args: Any) -> Any:
-        def inner(*args: Any) -> Any:
+    def _array_namespace_overload(*_args: Any) -> Any:
+        def inner(*_args: Any) -> Any:
             return np
 
         return inner
@@ -41,6 +41,13 @@ STR_TO_IS_NAMESPACE = {
 }
 
 
+def _identity[**P, T](
+    f: Callable[P, T],
+    /,
+) -> Callable[P, T]:
+    return f
+
+
 def _default_decorator(
     module: ModuleType,
     /,
@@ -49,20 +56,19 @@ def _default_decorator(
         import jax
 
         return jax.jit
-    elif is_numpy_namespace(module) or is_cupy_namespace(module):
+    if is_numpy_namespace(module) or is_cupy_namespace(module):
         # import numba
 
         # return numba.jit()
         # The success rate of numba.jit is low
-        return lambda x: x
-    elif is_torch_namespace(module):
+        return _identity
+    if is_torch_namespace(module):
         import torch
 
         return torch.compile
-    elif is_dask_namespace(module):
-        return lambda x: x
-    else:
-        return getattr(module, "jit", lambda x: x)
+    if is_dask_namespace(module):
+        return _identity
+    return getattr(module, "jit", _identity)
 
 
 Decorator = Callable[[Callable[Pin, Tin]], Callable[Pin, Tin]]
@@ -120,7 +126,9 @@ def jit(
 
     """
 
-    def new_decorator(f: Callable[Pinner, Tinner]) -> Callable[Pinner, Tinner]:
+    def new_decorator(
+        f: Callable[Pinner, Tinner],
+    ) -> Callable[Pinner, Tinner]:
         decorator_args_ = decorator_args or {}
         decorator_kwargs_ = decorator_kwargs or {}
         decorator_ = decorator or {}
@@ -130,14 +138,10 @@ def jit(
             for name_, is_namespace in STR_TO_IS_NAMESPACE.items():
                 if is_namespace(xp):
                     name = name_
-            else:
-                name = xp.__name__.split(".")[0]
+            name = xp.__name__.split(".")[0]
             decorator_args__ = decorator_args_.get(name, ())
             decorator_kwargs__ = decorator_kwargs_.get(name, {})
-            if name in decorator_:
-                decorator_current = decorator_[name]
-            else:
-                decorator_current = _default_decorator(xp)
+            decorator_current = decorator_[name] if name in decorator_ else _default_decorator(xp)
             try:
                 return decorator_current(f, *decorator_args__, **decorator_kwargs__)
             except Exception as e:

{array_api_jit-1.0.0 → array_api_jit-1.1.0/src/array_api_jit.egg-info}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: array-api-jit
-Version: 1.0.0
+Version: 1.1.0
 Summary: JIT decorator supporting multiple array API compatible libraries
-Author-email: 34j <34j.95a2p@simplelogin.com>
-License: MIT
+Author: 34j
+License-Expression: MIT
 Project-URL: Bug Tracker, https://github.com/34j/array-api-jit/issues
 Project-URL: Changelog, https://github.com/34j/array-api-jit/blob/main/CHANGELOG.md
 Project-URL: documentation, https://array-api-jit.readthedocs.io
@@ -12,12 +12,12 @@ Classifier: Development Status :: 2 - Pre-Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Natural Language :: English
 Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Software Development :: Libraries
-Requires-Python: >=3.10
+Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: array-api-compat>=1.11.2
@@ -32,19 +32,19 @@ Dynamic: license-file
   <a href="https://array-api-jit.readthedocs.io">
     <img src="https://img.shields.io/readthedocs/array-api-jit.svg?logo=read-the-docs&logoColor=fff&style=flat-square" alt="Documentation Status">
   </a>
+</p>
+<p align="center">
   <a href="https://codecov.io/gh/34j/array-api-jit">
     <img src="https://img.shields.io/codecov/c/github/34j/array-api-jit.svg?logo=codecov&logoColor=fff&style=flat-square" alt="Test coverage percentage">
   </a>
-</p>
-<p align="center">
   <a href="https://github.com/astral-sh/uv">
     <img src="https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/uv/main/assets/badge/v0.json" alt="uv">
   </a>
   <a href="https://github.com/astral-sh/ruff">
     <img src="https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json" alt="Ruff">
   </a>
-  <a href="https://github.com/pre-commit/pre-commit">
-    <img src="https://img.shields.io/badge/pre--commit-enabled-brightgreen?logo=pre-commit&logoColor=white&style=flat-square" alt="pre-commit">
+  <a href="https://github.com/j178/prek">
+    <img src="https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/j178/prek/master/docs/assets/badge-v0.json" alt="prek">
   </a>
 </p>
 <p align="center">
@@ -78,6 +78,8 @@ pip install array-api-jit
 Simply decorate your function with `@jit()`:
 
 ```python
+from typing import Any
+
 from array_api_jit import jit
 
 
@@ -112,25 +114,3 @@ def sin_n_times(x: Any, n: int) -> Any:
         x = xp.sin(x)
     return x
 ```
-
-## Contributors ✨
-
-Thanks goes to these wonderful people ([emoji key](https://allcontributors.org/docs/en/emoji-key)):
-
-<!-- prettier-ignore-start -->
-<!-- ALL-CONTRIBUTORS-LIST:START - Do not remove or modify this section -->
-<!-- markdownlint-disable -->
-<!-- markdownlint-enable -->
-<!-- ALL-CONTRIBUTORS-LIST:END -->
-<!-- prettier-ignore-end -->
-
-This project follows the [all-contributors](https://github.com/all-contributors/all-contributors) specification. Contributions of any kind welcome!
-
-## Credits
-
-[![Copier](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/copier-org/copier/master/img/badge/badge-grayscale-inverted-border-orange.json)](https://github.com/copier-org/copier)
-
-This package was created with
-[Copier](https://copier.readthedocs.io/) and the
-[browniebroke/pypackage-template](https://github.com/browniebroke/pypackage-template)
-project template.

{array_api_jit-1.0.0 → array_api_jit-1.1.0}/src/array_api_jit.egg-info/SOURCES.txt

@@ -1,7 +1,6 @@
 LICENSE
 README.md
 pyproject.toml
-setup.py
 src/array_api_jit/__init__.py
 src/array_api_jit/_main.py
 src/array_api_jit/py.typed

{array_api_jit-1.0.0 → array_api_jit-1.1.0}/tests/test_main.py

@@ -12,19 +12,19 @@ from numba.extending import overload
 
 from array_api_jit import jit
 
-IS_CI = environ.get("CI", "false").lower() in ("true", "1", "yes")
+IS_CI = environ.get("CI", "false").lower() in {"true", "1", "yes"}
 
 
 @overload(np.stack)
 def stack(arrays, axis=0):
     def inner(arrays, axis=0):
         if axis == 0:
-            shape = (len(arrays),) + arrays[0].shape  # noqa: RUF005
+            shape = (len(arrays),) + arrays[0].shape
             stacked_array = np.empty(shape, dtype=arrays[0].dtype)
             for j in prange(len(arrays)):
                 stacked_array[j] = arrays[j]
         elif axis == -1:
-            shape = arrays[0].shape + (len(arrays),)  # noqa: RUF005
+            shape = arrays[0].shape + (len(arrays),)
             stacked_array = np.empty(shape, dtype=arrays[0].dtype)
             for j in prange(len(arrays)):
                 stacked_array[..., j] = arrays[j]
@@ -104,7 +104,7 @@ legendre_assign_jit = jit({"numpy": numba.jit(nogil=True, parallel=True)})(legen
     IS_CI and platform != "linux", reason="Compiler not available on GitHub Actions"
 )
 def test_jit(xp: Any) -> None:
-    t = {}
+    t: dict[str, float] = {}
     for name, func in [("nojit", legendre), ("jit", legendre_jit)] + (
         [
             ("assign-nojit", legendre_assign),
@@ -113,16 +113,13 @@ def test_jit(xp: Any) -> None:
         if "jax" not in xp.__name__
         else []
     ):
-        print(xp.__name__, name)
-        for i in range(20):
+        for _ in range(20):
             with timer() as timer_:
                 x = xp.arange(1000000, dtype=xp.float32) / 1000000
                 p = func(x, 10)
                 assert p.shape == (1000000, 10)
+            assert timer_.elapsed is not None
             t[name] = timer_.elapsed
-            if i == 0:
-                print(f"First call: {timer_.elapsed:g}s")
-        print(f"Last call: {timer_.elapsed:g}s")
     if "numpy" not in xp.__name__:
         assert t["jit"] < t["nojit"], (
             f"JIT time {t['jit']} should be less than non-JIT time {t['nojit']}"

array_api_jit-1.0.0/pyproject.toml

@@ -1,162 +0,0 @@
-[build-system]
-build-backend = "setuptools.build_meta"
-requires = [ "setuptools" ]
-
-[project]
-name = "array-api-jit"
-version = "1.0.0"
-description = "JIT decorator supporting multiple array API compatible libraries"
-readme = "README.md"
-license = { text = "MIT" }
-authors = [
-  { name = "34j", email = "34j.95a2p@simplelogin.com" },
-]
-requires-python = ">=3.10"
-classifiers = [
-  "Development Status :: 2 - Pre-Alpha",
-  "Intended Audience :: Developers",
-  "Natural Language :: English",
-  "Operating System :: OS Independent",
-  "Programming Language :: Python :: 3.10",
-  "Programming Language :: Python :: 3.11",
-  "Programming Language :: Python :: 3.12",
-  "Programming Language :: Python :: 3.13",
-  "Topic :: Software Development :: Libraries",
-]
-
-dependencies = [
-    "array-api-compat>=1.11.2",
-]
-urls."Bug Tracker" = "https://github.com/34j/array-api-jit/issues"
-urls.Changelog = "https://github.com/34j/array-api-jit/blob/main/CHANGELOG.md"
-urls.documentation = "https://array-api-jit.readthedocs.io"
-urls.repository = "https://github.com/34j/array-api-jit"
-
-[dependency-groups]
-dev = [
-  "cm-time>=0.1.2",
-  "jax>=0.4.30",
-  "numba>=0.60.0",
-  "pytest>=8,<9",
-  "pytest-cov>=6,<7",
-  "torch>=2.7.1",
-]
-docs = [
-  "furo>=2023.5.20; python_version>='3.11'",
-  "myst-parser>=0.16; python_version>='3.11'",
-  "sphinx>=4; python_version>='3.11'",
-  "sphinx-autobuild>=2024,<2025; python_version>='3.11'",
-]
-
-[tool.ruff]
-format.docstring-code-format = true
-line-length = 100
-lint.select = [
-  "B",   # flake8-bugbear
-  "D",   # flake8-docstrings
-  "C4",  # flake8-comprehensions
-  "S",   # flake8-bandit
-  "F",   # pyflake
-  "E",   # pycodestyle
-  "W",   # pycodestyle
-  "UP",  # pyupgrade
-  "I",   # isort
-  "RUF", # ruff specific
-]
-lint.ignore = [
-  "D203", # 1 blank line required before class docstring
-  "D212", # Multi-line docstring summary should start at the first line
-  "D100", # Missing docstring in public module
-  "D104", # Missing docstring in public package
-  "D107", # Missing docstring in `__init__`
-  "D401", # First line of docstring should be in imperative mood
-]
-lint.per-file-ignores."conftest.py" = [ "D100" ]
-lint.per-file-ignores."docs/conf.py" = [ "D100" ]
-lint.per-file-ignores."setup.py" = [ "D100" ]
-lint.per-file-ignores."tests/**/*" = [
-  "D100",
-  "D101",
-  "D102",
-  "D103",
-  "D104",
-  "S101",
-]
-lint.isort.known-first-party = [ "array_api_jit", "tests" ]
-
-[tool.pytest.ini_options]
-addopts = """\
-    -v
-    -Wdefault
-    """
-pythonpath = [ "src" ]
-
-[tool.coverage.run]
-branch = true
-
-[tool.coverage.report]
-exclude_lines = [
-  "pragma: no cover",
-  "@overload",
-  "if TYPE_CHECKING",
-  "raise NotImplementedError",
-  'if __name__ == "__main__":',
-]
-
-[tool.mypy]
-check_untyped_defs = true
-disallow_any_generics = true
-disallow_incomplete_defs = true
-disallow_untyped_defs = true
-mypy_path = "src/"
-no_implicit_optional = true
-show_error_codes = true
-warn_unreachable = true
-warn_unused_ignores = true
-exclude = [
-  'docs/.*',
-  'setup.py',
-]
-
-[[tool.mypy.overrides]]
-module = "tests.*"
-allow_untyped_defs = true
-
-[[tool.mypy.overrides]]
-module = "docs.*"
-ignore_errors = true
-
-[tool.semantic_release]
-version_toml = [ "pyproject.toml:project.version" ]
-version_variables = [
-  "src/array_api_jit/__init__.py:__version__",
-  "docs/conf.py:release",
-]
-build_command = """
-pip install uv
-uv lock
-git add uv.lock
-uv build
-"""
-
-[tool.semantic_release.changelog]
-exclude_commit_patterns = [
-  '''chore(?:\([^)]*?\))?: .+''',
-  '''ci(?:\([^)]*?\))?: .+''',
-  '''refactor(?:\([^)]*?\))?: .+''',
-  '''style(?:\([^)]*?\))?: .+''',
-  '''test(?:\([^)]*?\))?: .+''',
-  '''build\((?!deps\): .+)''',
-  '''Merged? .*''',
-  '''Initial [Cc]ommit.*''', # codespell:ignore
-]
-
-[tool.semantic_release.changelog.environment]
-keep_trailing_newline = true
-
-[tool.semantic_release.branches.main]
-match = "main"
-
-[tool.semantic_release.branches.noop]
-match = "(?!main$)"
-prerelease = true

array_api_jit-1.0.0/setup.py

@@ -1,9 +0,0 @@
-#!/usr/bin/env python
-
-# This is a shim to allow GitHub to detect the package, build is done with uv
-# Taken from https://github.com/Textualize/rich
-
-import setuptools
-
-if __name__ == "__main__":
-    setuptools.setup(name="array-api-jit")