forked from enviPath/enviPy
[Fix] Adjust Rules according to manual QA, Add missing Rule (#452)
Co-authored-by: Tim Lorsbach <tim@lorsba.ch> Reviewed-on: enviPath/enviPy#452
This commit is contained in:
@ -1080,19 +1080,23 @@ def package_model(request, package_uuid, model_uuid):
|
||||
for pr in pred_res:
|
||||
if len(pr) > 0:
|
||||
products = []
|
||||
|
||||
for prod_set in pr.product_sets:
|
||||
logger.debug(f"Checking {prod_set}")
|
||||
products.append(tuple([x for x in prod_set]))
|
||||
|
||||
res["pred"].append(
|
||||
{
|
||||
"products": list(set(products)),
|
||||
"probability": pr.probability,
|
||||
"btrule": {k: getattr(pr.rule, k) for k in ["url", "name"]}
|
||||
if pr.rule is not None
|
||||
else None,
|
||||
}
|
||||
)
|
||||
products = list(set(products))
|
||||
|
||||
for prod in products:
|
||||
res["pred"].append(
|
||||
{
|
||||
"products": list(prod),
|
||||
"probability": pr.probability,
|
||||
"btrule": {k: getattr(pr.rule, k) for k in ["url", "name"]}
|
||||
if pr.rule is not None
|
||||
else None,
|
||||
}
|
||||
)
|
||||
|
||||
# Sort data by prob desc
|
||||
res["pred"] = sorted(
|
||||
|
||||
Reference in New Issue
Block a user