From de48aa5b348898f2f59ba78f83f8a0103f25fd94 Mon Sep 17 00:00:00 2001 From: "Michael J. Williams" Date: Tue, 27 Oct 2026 12:39:38 +0100 Subject: [PATCH] BUG(array-api): type namespace predicate mapping --- src/sequince/array_api.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/sequince/array_api.py b/src/sequince/array_api.py index 80881b0e..34b268c2 100644 --- a/src/sequince/array_api.py +++ b/src/sequince/array_api.py @@ -1,5 +1,6 @@ from __future__ import annotations +from collections.abc import Callable from typing import Any import numpy as _np @@ -23,7 +24,7 @@ from ._scipy_compat import enable_scipy_array_api as _enable_scipy_array_api from .typing import Array -IS_NAMESPACE_FUNCTIONS = { +IS_NAMESPACE_FUNCTIONS: dict[str, Callable[[Any], bool]] = { "numpy": is_numpy_namespace, "torch": is_torch_namespace, "jax": is_jax_namespace,