Merge pull request #16 from Alex-Lekov/develop

Fix y_train
Alex-Lekov · Aug 5, 2020 · d4198df · d4198df
2 parents 67c5907 + 1bcf476
commit d4198df
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -5,6 +5,11 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
 
 
+## [0.07.26]
+
+### Fix
+- if y_train is not pd.DataFrame
+
 ## [0.07.26]
 
 ### Add

diff --git a/automl_alex/__version__.py b/automl_alex/__version__.py
@@ -1 +1 @@
-__version__ = "0.07.26"
+__version__ = "0.08.05"
diff --git a/automl_alex/databunch.py b/automl_alex/databunch.py
@@ -70,7 +70,7 @@ def __init__(self,
                 self.X_test_source = pd.DataFrame(X_test)
 
         if y_train is not None:
-            self.y_train = y_train
+            self.y_train = pd.DataFrame(y_train)
         else:
             raise Exception("No target data!")