Commit 0fbe46e7 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Do not automatically wrap pipeline in dask wrappers to allow cusotm pipelines

parent 1b803d1d
......@@ -89,9 +89,7 @@ def vanilla_pad(ctx, pipeline, database, dask_client, groups, output, checkpoint
["checkpoint"], pipeline, features_dir=output, model_path=output
)
if dask_client is not None:
pipeline = mario.wrap(["dask"], pipeline)
else:
if dask_client is None:
logger.warning("`dask_client` not set. Your pipeline will run locally")
# create an experiment info file
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment