From ed74a51086a612c36e20c7232abb55eab4e0fbbc Mon Sep 17 00:00:00 2001 From: eddiebergman Date: Tue, 1 Aug 2023 17:14:19 +0200 Subject: [PATCH] fix: __contains__ for configspace --- ConfigSpace/configuration_space.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/ConfigSpace/configuration_space.py b/ConfigSpace/configuration_space.py index 7f7a7953..29f624f6 100644 --- a/ConfigSpace/configuration_space.py +++ b/ConfigSpace/configuration_space.py @@ -1140,6 +1140,9 @@ def __getitem__(self, key: str) -> Hyperparameter: return hp + def __contains__(self, key: str) -> bool: + return key in self._hyperparameters + def __repr__(self) -> str: retval = io.StringIO() retval.write("Configuration space object:\n Hyperparameters:\n")