change model to id parameter (#161)

Open-EO · Apr 19, 2022 · ee54cad · ee54cad
1 parent 2bcc090
commit ee54cad
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/openeo_processes/cubes.py b/src/openeo_processes/cubes.py
@@ -2142,11 +2142,11 @@ def load_ml_model():
 class LoadMLModel:
 
     @staticmethod
-    def exec_num(model, input_filepath = 'path'):
-        date = os.listdir(f'{input_filepath}/jobs/{model}')
+    def exec_num(id, input_filepath = 'path'):
+        date = os.listdir(f'{input_filepath}/jobs/{id}')
         if len(date) > 0:
             date = date[0]
-        filepath = f'{input_filepath}/jobs/{model}/{date}/result/out_model.json'
+        filepath = f'{input_filepath}/jobs/{id}/{date}/result/out_model.json'
         model_xgb = xgb.Booster()
         model_xgb.load_model(filepath)
         return model_xgb