From d2a4181f897633bfa6c538f54f315568abf65f8c Mon Sep 17 00:00:00 2001
From: "Patrick J. Roddy" <patrickjamesroddy@gmail.com>
Date: Wed, 30 Apr 2025 17:07:38 +0100
Subject: [PATCH] gh-612: handle NoneType in `get_namespace`

---
 glass/_array_api_utils.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/glass/_array_api_utils.py b/glass/_array_api_utils.py
index 5f8fa992..10284068 100644
--- a/glass/_array_api_utils.py
+++ b/glass/_array_api_utils.py
@@ -16,7 +16,11 @@ def get_namespace(*arrays: NDArray[Any] | Array) -> ModuleType:
     if they do not.
     """
     namespace = arrays[0].__array_namespace__()
-    if not all(array.__array_namespace__() == namespace for array in arrays):
+    if any(
+        array.__array_namespace__() != namespace
+        for array in arrays
+        if array is not None
+    ):
         msg = "input arrays should belong to the same array library"
         raise ValueError(msg)