diff --git a/epdb/management/commands/bootstrap.py b/epdb/management/commands/bootstrap.py index 5347eabc..4beb3222 100644 --- a/epdb/management/commands/bootstrap.py +++ b/epdb/management/commands/bootstrap.py @@ -187,17 +187,17 @@ class Command(BaseCommand): # Create RR ml_model = MLRelativeReasoning.create( - pack, - 'ECC - BBD - T0.5', - 'ML Relative Reasoning', - [mapping['EAWAG-BBD']], - [mapping['EAWAG-BBD']], - [], - 0.5 + package=pack, + rule_packages=[mapping['EAWAG-BBD']], + data_packages=[mapping['EAWAG-BBD']], + eval_packages=[], + threshold=0.5, + name='ECC - BBD - T0.5', + description='ML Relative Reasoning', ) - X, y = ml_model.build_dataset() - ml_model.build_model(X, y) + ml_model.build_dataset() + ml_model.build_model() # ml_model.evaluate_model() # If available create EnviFormerModel diff --git a/tests/test_model.py b/tests/test_model.py index f1652b03..0b63a35c 100644 --- a/tests/test_model.py +++ b/tests/test_model.py @@ -37,12 +37,12 @@ class ModelTest(TestCase): mod = MLRelativeReasoning.create( self.package, - 'ECC - BBD - 0.5', - 'Created MLRelativeReasoning in Testcase', rule_package_objs, data_package_objs, eval_packages_objs, - threshold + threshold, + 'ECC - BBD - 0.5', + 'Created MLRelativeReasoning in Testcase', ) ds = mod.load_dataset()