fix estimate bug

Signed-off-by: Yanbo0101 <[email protected]>
sustainable-computing-io · Aug 28, 2023 · ad68aa6 · ad68aa6
1 parent 0bcf63a
commit ad68aa6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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