Make forge_experiment's model_init_params optional

HunterMcGushion · Jul 14, 2019 · d811cfd · d811cfd
1 parent 0b40487
commit d811cfd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hyperparameter_hunter/optimization/protocol_core.py b/hyperparameter_hunter/optimization/protocol_core.py
@@ -216,7 +216,7 @@ def __init__(
     def forge_experiment(
         self,
         model_initializer,
-        model_init_params,
+        model_init_params=None,
         model_extra_params=None,
         feature_engineer=None,
         feature_selector=None,
@@ -263,8 +263,8 @@ def forge_experiment(
         differences between calling :meth:`forge_experiment` and instantiating
         :class:`~hyperparameter_hunter.experiments.CVExperiment`"""
         self.model_initializer = model_initializer
-
         self.model_init_params = identify_algorithm_hyperparameters(self.model_initializer)
+        model_init_params = model_init_params if model_init_params is not None else {}
         try:
             self.model_init_params.update(model_init_params)
         except TypeError: