fix bugs stable now

KohlerHECTOR · Jul 18, 2024 · e5bb8cf · e5bb8cf
1 parent 6e88603
commit e5bb8cf
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -7,7 +7,7 @@ It is also possible to do regression, including multi-output regression.
 
 ## Installation
 ```bash
-pip install git+https://github.com/KohlerHECTOR/dpdt-py.git@v0.1.7
+pip install git+https://github.com/KohlerHECTOR/dpdt-py.git@v0.2.1
 ```
 
 

diff --git a/docs/conf.py b/docs/conf.py
@@ -9,7 +9,7 @@
 project = "dpdt"
 copyright = "2024, Hector Kohler"
 author = "Hector Kohler"
-release = "0.2.0"
+release = "0.2.1"
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration

diff --git a/docs/usage.md b/docs/usage.md
@@ -1,6 +1,6 @@
 ## Installation
 ```bash
-pip install git+https://github.com/KohlerHECTOR/dpdt-py.git@v0.1.7
+pip install git+https://github.com/KohlerHECTOR/dpdt-py.git@v0.2.1
 ```
 
 

diff --git a/dpdt/dpdt_sklearn.py b/dpdt/dpdt_sklearn.py
@@ -698,7 +698,7 @@ def fit(self, X, y, feature_costs=None):
 
         # Store the training data to predict later
         self.X_ = X
-        self.y_ = y
+        self.y_ = y.astype(float)
 
         if self.max_nb_trees < 2:
             self.zetas_ = np.zeros(1)

diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 import os
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 
 packages = find_packages(
     exclude=[