From 3f877c0dbb259e538ac521af6802c9a89c6507b7 Mon Sep 17 00:00:00 2001 From: Patrick Kidger <33688385+patrick-kidger@users.noreply.github.com> Date: Tue, 8 Nov 2022 22:52:17 -0800 Subject: [PATCH] Update array_types.py (#41) --- jaxtyping/array_types.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/jaxtyping/array_types.py b/jaxtyping/array_types.py index c96a5f0..32790c4 100644 --- a/jaxtyping/array_types.py +++ b/jaxtyping/array_types.py @@ -383,15 +383,15 @@ class _MetaAbstractDtype(type): elem = compile(elem, "", "eval") elem = _SymbolicDim(elem, broadcastable) dims.append(elem) + # In python 3.8, e.g., typing.Union lacks `__name__`. + try: + type_str = array_type.__name__ + except AttributeError: + type_str = repr(array_type) if _array_name_format == "dtype_and_shape": - # In python 3.8, e.g., typing.Union lacks `__name__`. - try: - type_str = array_type.__name__ - except AttributeError: - type_str = repr(array_type) name = f"{cls.__name__}[{type_str}, '{dim_str}']" elif _array_name_format == "array": - name = array_type.__name__ + name = type_str else: raise ValueError(f"array_name_format {_array_name_format} not recognised") out = _MetaAbstractArray(