diff --git a/fugue_incubator_version/__init__.py b/fugue_incubator_version/__init__.py index 81f0fde..b1a19e3 100644 --- a/fugue_incubator_version/__init__.py +++ b/fugue_incubator_version/__init__.py @@ -1 +1 @@ -__version__ = "0.0.4" +__version__ = "0.0.5" diff --git a/fugue_tune/sklearn.py b/fugue_tune/sklearn.py index 066dcf4..3b9387f 100644 --- a/fugue_tune/sklearn.py +++ b/fugue_tune/sklearn.py @@ -36,10 +36,10 @@ def suggest_sk_model( partition_keys: List[str] = _EMPTY_LIST, top_n: int = 1, objective_runner: Optional[ObjectiveRunner] = None, - distributable: bool = False, - engine: Any = NativeExecutionEngine, + distributable: Optional[bool] = None, + execution_engine: Any = NativeExecutionEngine, ) -> List[Dict[str, Any]]: - e = to_instance(engine, ExecutionEngine) + e = to_instance(execution_engine, ExecutionEngine) model_path = serialize_path if save_model else "" dag = FugueWorkflow()