fix: __contains__ for configspace

automl · Aug 1, 2023 · ed74a51 · ed74a51
1 parent 93abc5b
commit ed74a51
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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")