diff --git a/cmd/main.py b/cmd/main.py index 7606a054..1a558cef 100644 --- a/cmd/main.py +++ b/cmd/main.py @@ -387,7 +387,7 @@ def estimate(args): if pipeline_metadata is None: print("no metadata for pipeline {}.".format(pipeline_name)) continue - pipeline = get_pipeline(pipeline_name, args.profile, pipeline_metadata["isolator"], pipeline_metadata["abs_trainers"], pipeline_metadata["dyn_trainers"], args.energy_source, valid_fg) + pipeline = get_pipeline(pipeline_name, args.extractor, args.profile, pipeline_metadata["isolator"], pipeline_metadata["abs_trainers"], pipeline_metadata["dyn_trainers"], args.energy_source, valid_fg) if pipeline is None: print("cannot get pipeline {}.".format(pipeline_name)) continue