Merge pull request #39 from alliander-opensource/bugfix-predictionloop

Bugfix predictionloop
OpenSTEF · Mar 9, 2021 · c66a7ab · c66a7ab
2 parents cd83d09 + 0b3cb2f
commit c66a7ab
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/openstf/model/capacity/model.py b/openstf/model/capacity/model.py
@@ -83,10 +83,12 @@ def save(self, directory):
             os.makedirs(directory, exist_ok=True)
 
         # save model
-        pickle.dump(self.model, open(directory / "model.pkl", "wb"))
+        with open(directory / "model.pkl", "wb") as fh:
+            pickle.dump(self.model, fh)
 
         # save classes
-        pickle.dump(self.classes, open(directory / "classes.pkl", "wb"))
+        with open(directory / "classes.pkl", "wb") as fh:
+            pickle.dump(self.classes, fh)
 
     def load(self, directory):
         # load model

diff --git a/openstf/tasks/utils/predictionjobloop.py b/openstf/tasks/utils/predictionjobloop.py
@@ -77,7 +77,7 @@ def _get_prediction_jobs(self):
 
         return prediction_jobs
 
-    def _add_missing_info_prediciton_jobs(self, prediction_jobs_input):
+    def _add_missing_info_prediction_jobs(self, prediction_jobs_input):
         prediction_jobs = []
 
         for pj in prediction_jobs_input:

diff --git a/setup.py b/setup.py
@@ -28,7 +28,7 @@ def read_long_description_from_readme():
 
 setup(
     name="openstf",
-    version="1.5.3",
+    version="1.5.4",
     packages=find_packages(include=["openstf", "openstf.*"]),
     description="Open short term forcasting",
     long_description=read_long_description_from_readme(),