From e61a37f0a37f7c67c83daf816e161a69799dc479 Mon Sep 17 00:00:00 2001 From: Patrick Kidger <33688385+patrick-kidger@users.noreply.github.com> Date: Thu, 15 Sep 2022 16:05:49 +0100 Subject: [PATCH] Fix for boolean checking not working (#19) --- jaxtyping/array_types.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jaxtyping/array_types.py b/jaxtyping/array_types.py index 631c281..ae5874e 100644 --- a/jaxtyping/array_types.py +++ b/jaxtyping/array_types.py @@ -446,7 +446,7 @@ if TYPE_CHECKING: from typing_extensions import Annotated as UInt32 from typing_extensions import Annotated as UInt64 else: - _bool = "bool" + _bool = "bool_" _uint8 = "uint8" _uint16 = "uint16" _uint32 = "uint32"