roll back the config check

microsoft · Jan 10, 2024 · 0cc11e8 · 0cc11e8
1 parent 7a68ce1
commit 0cc11e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mlos_core/mlos_core/optimizers/optimizer.py b/mlos_core/mlos_core/optimizers/optimizer.py
@@ -136,8 +136,8 @@ def suggest(self, context: Optional[pd.DataFrame] = None, defaults: bool = False
                 "Suggest returned a configuration with the wrong number of parameters."
         if self._space_adapter:
             configuration = self._space_adapter.transform(configuration)
-            assert set(configuration.columns) == set(self.parameter_space), \
-                "Space adapter transformed configuration has incorrect parameters."
+            assert set(configuration.columns).issubset(set(self.parameter_space)), \
+                "Space adapter transformed configuration with the wrong number of parameters."
         return configuration
 
     @abstractmethod