Merge pull request #11 from fugue-project/0.0.4.1

fine tune sklearn interface
fugue-project · Jan 3, 2021 · fecf328 · fecf328
2 parents 61de323 + e318726
commit fecf328
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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()