feat: do not use is_array_container

Author: alexfikl

arraycontext/container/traversal.py

@@ -1095,8 +1095,9 @@ def _flat_size(subary: ArrayOrContainerOrScalar) -> Array | int | np.integer:
         try:
             iterable = serialize_container(subary)
         except NotAnArrayContainerError:
-            assert not is_array_container(subary)
-            assert not is_scalar_like(subary)
+            if TYPE_CHECKING:
+                assert not is_array_container(subary)
+                assert not is_scalar_like(subary)
 
             if common_dtype is None:
                 common_dtype = subary.dtype