Skip to content

Commit fc86bf2

Browse files
author
Sunil Thaha
authored
Merge pull request #516 from vprashar2929/chore-tp
fix: check length of PROM_THIRDPARTY_METRICS
2 parents 9b4b832 + c6acd08 commit fc86bf2

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/kepler_model/cmd/main.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,7 @@ def query(args):
145145
queries = None
146146
if args.thirdparty_metrics != "":
147147
queries = [m for m in available_metrics if args.metric_prefix in m or m in args.thirdparty_metrics]
148-
elif [""] != PROM_THIRDPARTY_METRICS:
148+
elif len(PROM_THIRDPARTY_METRICS) > 0:
149149
queries = [m for m in available_metrics if args.metric_prefix in m or m in PROM_THIRDPARTY_METRICS]
150150
else:
151151
queries = [m for m in available_metrics if args.metric_prefix in m]
@@ -210,7 +210,7 @@ def extract(args):
210210
# Inject thirdparty_metrics to FeatureGroup
211211
if args.thirdparty_metrics != "":
212212
update_thirdparty_metrics(args.thirdparty_metrics)
213-
elif [""] != PROM_THIRDPARTY_METRICS:
213+
elif len(PROM_THIRDPARTY_METRICS) > 0:
214214
update_thirdparty_metrics(PROM_THIRDPARTY_METRICS)
215215
valid_fg = get_valid_feature_group_from_queries([query for query in query_results.keys() if len(query_results[query]) > 1])
216216
ot, fg = check_ot_fg(args, valid_fg)
@@ -328,7 +328,7 @@ def train_from_data(args):
328328
# Inject thirdparty_metrics to FeatureGroup
329329
if args.thirdparty_metrics != "":
330330
update_thirdparty_metrics(args.thirdparty_metrics)
331-
elif [""] != PROM_THIRDPARTY_METRICS:
331+
elif len(PROM_THIRDPARTY_METRICS) > 0:
332332
update_thirdparty_metrics(PROM_THIRDPARTY_METRICS)
333333
valid_fg = [fg_key for fg_key in FeatureGroups.keys()]
334334
ot, fg = check_ot_fg(args, valid_fg)
@@ -421,7 +421,7 @@ def train(args):
421421
# Inject thirdparty_metrics to FeatureGroup
422422
if args.thirdparty_metrics != "":
423423
update_thirdparty_metrics(args.thirdparty_metrics)
424-
elif [""] != PROM_THIRDPARTY_METRICS:
424+
elif len(PROM_THIRDPARTY_METRICS) > 0:
425425
update_thirdparty_metrics(PROM_THIRDPARTY_METRICS)
426426

427427
pipeline_name = default_train_output_pipeline
@@ -564,7 +564,7 @@ def estimate(args):
564564
# Inject thirdparty_metrics to FeatureGroup
565565
if args.thirdparty_metrics != "":
566566
update_thirdparty_metrics(args.thirdparty_metrics)
567-
elif [""] != PROM_THIRDPARTY_METRICS:
567+
elif len(PROM_THIRDPARTY_METRICS) > 0:
568568
update_thirdparty_metrics(PROM_THIRDPARTY_METRICS)
569569

570570
inputs = args.input.split(",")

0 commit comments

Comments
 (0)