[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:
2026-08-13 17:37:38 +12:00
parent ada270aa3c
commit 2d3dca6a75
5 changed files with 140 additions and 17 deletions

View File

@ -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(