fixes bug in variation for ml_type variable

lacava · Jul 13, 2017 · 5a46411 · 5a46411
1 parent 6ba6f8b
commit 5a46411
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/few/_version.py b/few/_version.py
@@ -6,4 +6,4 @@
 
 """
 
-__version__ = '0.0.41'
+__version__ = '0.0.42'
diff --git a/few/variation.py b/few/variation.py
@@ -20,7 +20,7 @@ def variation(self,parents):
         """performs variation operators on parents."""
         # downselect to features that are important
         if (self.valid(parents) and
-            type(self.ml).__name__ != 'SVC' and type(self.ml).__name__ != 'SVR'):
+            self.ml_type != 'SVC' and self.ml_type != 'SVR'):
             # this is needed because svm has a bug that throws valueerror on
             # attribute check
             if hasattr(self.ml.named_steps['ml'],'coef_'):
@@ -151,7 +151,8 @@ def cross(self,p_i,p_j, max_depth = 2):
         #swap subtrees
         tmpi = p_i[:]
         tmpj = p_j[:]
-        tmpi[x_i_begin:x_i_end+1:],tmpj[x_j_begin:x_j_end+1:] = tmpj[x_j_begin:x_j_end+1:],tmpi[x_i_begin:x_i_end+1:]
+        tmpi[x_i_begin:x_i_end+1:],tmpj[x_j_begin:x_j_end+1:] = \
+                        tmpj[x_j_begin:x_j_end+1:],tmpi[x_i_begin:x_i_end+1:]
 
         if not self.is_valid_program(p_i) or not self.is_valid_program(p_j):
             # pdb.set_trace()