From 6f59ff9f49556c67122e8b1b6fd4e1c16c5b974e Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Sun, 26 Sep 2021 03:03:53 -0500
Subject: [PATCH] ArrayContainer fixes for numpy arrays as leaf classes
---
arraycontext/container/__init__.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/arraycontext/container/__init__.py b/arraycontext/container/__init__.py
index ea20a5a..53506a0 100644
--- a/arraycontext/container/__init__.py
+++ b/arraycontext/container/__init__.py
@@ -218,7 +218,11 @@ def is_array_container(ary: Any) -> bool:
"cheaper option, see is_array_container_type.",
DeprecationWarning, stacklevel=2)
return (serialize_container.dispatch(ary.__class__)
- is not serialize_container.__wrapped__) # type:ignore[attr-defined]
+ is not serialize_container.__wrapped__ # type:ignore[attr-defined]
+ # numpy values with scalar elements aren't array containers
+ and not (isinstance(ary, np.ndarray)
+ and ary.dtype.kind != "O")
+ )
@singledispatch
--
GitLab