adding sympy to meta.yml

scikit-learn-contrib · Aug 8, 2016 · 488c95e · 488c95e
1 parent 971d0a3
commit 488c95e
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 7 deletions.
diff --git a/conda-recipe/meta.yaml b/conda-recipe/meta.yaml
@@ -13,22 +13,25 @@ requirements:
     - numpy
     - scikit-learn
     - nose
-    - six  
+    - six
+    - sympy
   run:
     - python
     - numpy
     - scipy
     - scikit-learn
-    - six  
+    - six
+    - sympy
 
 test:
   requires:
     - numpy
     - scipy
     - scikit-learn
     - nose
+    - sympy
   imports:
-    - pyearth 
+    - pyearth
 
 about:
   home: https://github.com/scikit-learn-contrib/py-earth

diff --git a/pyearth/export.py b/pyearth/export.py
@@ -38,6 +38,7 @@ def export_python_string(earth_model, function_name="model"):
     """.format(function_name, ",\n\t\t".join(accessors))
 
 def export_sympy(earth_model):
+
     from sympy import Symbol, Add, Mul, Max, RealNumber, Piecewise, Pow, And
     """
     Exports model in Sympy specific syntax for use with Sympy (and conversion with Sympy Codegen)
@@ -115,10 +116,10 @@ def termify(bf, i):
                     terms.append(term)
 
             elif 'MissingnessBasisFunction' in bf_class:
-                print bf_var
+                print(bf_var)
 
             else:
-                print bf_var
+                print(bf_var)
 
             parent = parent.get_parent()
         term_list.append(terms)

diff --git a/pyearth/test/test_export.py b/pyearth/test/test_export.py
@@ -71,5 +71,3 @@ def test_export_sympy():
 
         y_pred = model.predict(X_df)
         assert_list_almost_equal(y_pred, y_pred_sympy)
-
-
Original file line number	Diff line number	Diff line change
Expand Up		@@ -71,5 +71,3 @@ def test_export_sympy():

		y_pred = model.predict(X_df)
		assert_list_almost_equal(y_pred, y_pred_sympy)