fix linter

matsim-vsp · Jun 15, 2024 · 0f535aa · 0f535aa
1 parent 7429274
commit 0f535aa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/matsim/scenariogen/ml/models.py b/matsim/scenariogen/ml/models.py
@@ -295,7 +295,7 @@ def model_to_java(name, package, model_and_error: tuple, scaler, bounds, df):
 def model_to_py(name, model_and_error, scaler, df):
     import m2cgen as m2c
 
-    code = m2c.export_to_python(model[0], function_name="score")
+    code = m2c.export_to_python(model_and_error[0], function_name="score")
 
     code, params = replace_params(code)
 

diff --git a/matsim/scenariogen/ml/train.py b/matsim/scenariogen/ml/train.py
@@ -195,7 +195,7 @@ def write_python(self, folder, name):
 
             # TODO: bounds not implemented
 
-            code = model_to_py(t, self.best, self.scaler, self.get(None)[0])
+            code = model_to_py(name, self.best, self.scaler, self.get(None)[0])
             f.write("# -*- coding: utf-8 -*-\n")
             f.write("\"\"\"%s\nError: %f\"\"\"\n" % self.best)
             f.write(code)