From a0c2dbf238c9877baf0235455ccee666e3011e56 Mon Sep 17 00:00:00 2001 From: Jeremiah Lowin <153965+jlowin@users.noreply.github.com> Date: Sun, 28 Apr 2024 23:06:02 -0400 Subject: [PATCH] Fix bool classifier --- src/marvin/_mappings/types.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/marvin/_mappings/types.py b/src/marvin/_mappings/types.py index a2f2cb804..f243a7b16 100644 --- a/src/marvin/_mappings/types.py +++ b/src/marvin/_mappings/types.py @@ -97,7 +97,7 @@ def cast_type_to_labels(type_: Union[type, GenericAlias]) -> list[str]: except Exception as exc: raise ValueError(f"Unable to cast type to labels: {exc}") elif type_ is bool: - return ["false", "true"], [False, True] + return ["false", "true"] else: raise TypeError(f"Expected Literal, Enum, bool, or list, got {type_}.")