2 files changed, 3 insertions, 1 deletions
diff --git a/tools/dataProvider.h b/tools/dataProvider.h
index ce2bf12..d311149 100644
--- a/tools/dataProvider.h
+++ b/tools/dataProvider.h
@@ -17,7 +17,7 @@
 
 typedef Eigen::MatrixXd DataSet;
 
-typedef std::vector<double> Labels;
+typedef Eigen::VectorXd Labels;
 
 class DataProvider  //Virtual base class for data input
 {
@@ -39,6 +39,7 @@ public:
     virtual int getDataSet(DataSet &out) = 0;
     virtual int getLabel(Labels &out) = 0;
     virtual int open()=0;
+    virtual int close()=0;
 };
 
 #endif
 \ No newline at end of file
diff --git a/tools/fileDataProvider.h b/tools/fileDataProvider.h
index 8a499ca..a4cc252 100644
--- a/tools/fileDataProvider.h
+++ b/tools/fileDataProvider.h
@@ -18,6 +18,7 @@ public:
         return 0;
     };
     virtual int open(){eof=true;return 0;};
+    virtual int close(){return 0;};
 };
 
 #endif
 \ No newline at end of file