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")