diff --git a/epdb/views.py b/epdb/views.py index 8d83b786..bbe984b2 100644 --- a/epdb/views.py +++ b/epdb/views.py @@ -917,7 +917,7 @@ def package_models(request, package_uuid): params["threshold"] = threshold mod = EnviFormer.create(**params) - elif model_type == "mlrr": + elif model_type == "ml-relative-reasoning": # ML Specific threshold = float(request.POST.get("model-threshold", 0.5)) # TODO handle additional fingerprinter @@ -941,7 +941,7 @@ def package_models(request, package_uuid): params["app_domain_local_compatibility_threshold"] = local_compatibility_threshold mod = MLRelativeReasoning.create(**params) - elif model_type == "rbrr": + elif model_type == "rule-based-relative-reasoning": params["rule_packages"] = [ PackageManager.get_package_by_url(current_user, p) for p in rule_packages ]