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

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
......
Markdown is supported
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