fix bug in subspace identification

flaml/tune/space.py

@@ -418,14 +418,13 @@ def indexof(domain: Dict, config: Dict) -> int:
         return domain.categories.index(config)
     # print(config)
     for i, cat in enumerate(domain.categories):
-        # print(cat)
         if not isinstance(cat, dict):
             continue
         # print(len(cat), len(config))
-        if len(cat) != len(config):
-            continue
+        # if len(cat) != len(config):
+        #     continue
         # print(cat.keys())
-        if not set(cat.keys()).issubset(set(config.keys())):
+        if not set(config.keys()).issubset(set(cat.keys())):
             continue
         # print(domain.const[i])
         if all(config[key] == value for key, value in domain.const[i].items()):

test/test_automl.py

@@ -646,6 +646,13 @@ class TestAutoML(unittest.TestCase):
         except ImportError:
             return
 
+    def test_parallel_classification(self):
+        from sklearn.datasets import make_classification
+
+        X, y = make_classification(1000, 10)
+        automl = AutoML()
+        automl.fit(X, y, time_budget=10, task="classification", n_concurrent_trials=2)
+
     def test_parallel_xgboost(self, hpo_method=None):
         automl_experiment = AutoML()
         automl_settings = {