From 21488fb06d39ef36acc6fa01a08cd2d627a8e77a Mon Sep 17 00:00:00 2001 From: eddiebergman Date: Tue, 16 Apr 2024 19:27:08 +0200 Subject: [PATCH] fix: Make sure to use `NotSet` in simple api --- src/ConfigSpace/api/types/categorical.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/ConfigSpace/api/types/categorical.py b/src/ConfigSpace/api/types/categorical.py index 54653c5e..900f112e 100644 --- a/src/ConfigSpace/api/types/categorical.py +++ b/src/ConfigSpace/api/types/categorical.py @@ -5,6 +5,7 @@ from typing_extensions import TypeAlias from ConfigSpace.hyperparameters import CategoricalHyperparameter, OrdinalHyperparameter +from ConfigSpace.types import NotSet, _NotSet # We only accept these types in `items` T: TypeAlias = Union[str, int, float] @@ -16,7 +17,7 @@ def Categorical( name: str, items: Sequence[T], *, - default: T | None = None, + default: T | _NotSet = NotSet, weights: Sequence[float] | None = None, ordered: Literal[False], meta: dict | None = None, @@ -29,7 +30,7 @@ def Categorical( name: str, items: Sequence[T], *, - default: T | None = None, + default: T | _NotSet = NotSet, weights: Sequence[float] | None = None, ordered: Literal[True], meta: dict | None = None, @@ -42,7 +43,7 @@ def Categorical( name: str, items: Sequence[T], *, - default: T | None = None, + default: T | _NotSet = NotSet, weights: Sequence[float] | None = None, ordered: bool = ..., meta: dict | None = None, @@ -53,7 +54,7 @@ def Categorical( name: str, items: Sequence[T], *, - default: T | None = None, + default: T | _NotSet = NotSet, weights: Sequence[float] | None = None, ordered: bool = False, meta: dict | None = None,