diff --git a/epdb/management/commands/create_ml_models.py b/epdb/management/commands/create_ml_models.py index 4f86c957..fa339966 100644 --- a/epdb/management/commands/create_ml_models.py +++ b/epdb/management/commands/create_ml_models.py @@ -93,7 +93,6 @@ class Command(BaseCommand): model = EnviFormer.create( pack, data_packages=data_packages, - eval_packages=eval_packages, threshold=options["threshold"], name=f"EnviFormer - {', '.join(options['data_packages'])} - T{options['threshold']:.2f}", description=f"EnviFormer transformer trained on {options['data_packages']} " @@ -104,7 +103,6 @@ class Command(BaseCommand): package=pack, rule_packages=rule_packages, data_packages=data_packages, - eval_packages=eval_packages, threshold=options["threshold"], name=f"ECC - {', '.join(options['data_packages'])} - T{options['threshold']:.2f}", description=f"ML Relative Reasoning trained on {options['data_packages']} with rules from "