Fix _is_param_distributions_all_tune_domains (#209)

ray-project · Jun 3, 2021 · 1697f9d · 1697f9d
1 parent b618b8e
commit 1697f9d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tune_sklearn/tune_search.py b/tune_sklearn/tune_search.py
@@ -474,7 +474,7 @@ def get_sample(dist):
             self.n_trials = min(self.n_trials, samples)
 
     def _is_param_distributions_all_tune_domains(self):
-        return all(
+        return isinstance(self.param_distributions, dict) and all(
             isinstance(v, Domain) for k, v in self.param_distributions.items())
 
     def _get_bohb_config_space(self):