Merge pull request #54 from alliander-opensource/bugfix-release-openstf

bugfix
OpenSTEF · Mar 24, 2021 · 1dc5a79 · 1dc5a79
2 parents 0d96516 + e08e795
commit 1dc5a79
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/openstf/model/train.py b/openstf/model/train.py
@@ -197,7 +197,7 @@ def preprocess_for_model_training(pj, context):
 
     featureset = context.database.get_featureset(hyperparams["featureset_name"])
     datetime_start = datetime.utcnow() - timedelta(
-        days=hyperparams["training_period_days"]
+        days=int(hyperparams["training_period_days"])
     )
     datetime_end = datetime.utcnow()
 
@@ -237,7 +237,7 @@ def train_model(model_trainer, split_input_data):
             validation and test
     """
     # Train model
-    model_trainer.train(split_input_data.train_data, split_input_data.validation_data)
+    model_trainer.train(split_input_data.train, split_input_data.validation)
 
 
 def predict_after_model_training(pj, model_trainer, split_input_data):

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