fixed calls to create MLRelativeReasoning during bootstrap.py and test_model.py

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

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