fixed setup.py for eigen path

Author: Leily Rabbani

setup.py

@@ -3,29 +3,23 @@
 from setuptools.command.build_ext import build_ext
 import sys
 import setuptools
+from sysconfig import get_config_var, get_paths
 
-__version__ = '0.0.0'
-
-
-class get_eigen_include(object):
-    """Includes eigen3 path """
 
-    def __init__(self, user=False):
-        self.user = user
-
-    def __str__(self):
-        import eigen3
-        return eigen3.get_include(self.user)
+__version__ = '0.0.0'
 
+def get_include(): #TODO 
+	info = get_paths()
+	Eigen_path = info['include']+'/../eigen3'
+	return Eigen_path      
 
 KR_module = [
     Extension(
         'KRBalancing',
         ['src/KRBalancing.cpp'],
         include_dirs=[
             # Path to eigen3 headers
-            get_eigen_include(),
-            get_eigen_include(user=True)
+            get_include(),
         ],
         language='c++'
     ),

src/KRBalancing.cpp

@@ -137,7 +137,7 @@ class kr_balancing{
         }//End of the inner 'while'
       }
       SparseMatrixR  get_output(){
-        std::cout << "export output"<<std::endl;
+        std::cout << "export output of shape "<< A.cwiseProduct(x*x.transpose()).rows() <<std::endl;
         return A.cwiseProduct(x*x.transpose());
       }
      private: