Don't forget to add the full training path

src/main/java/weka/finito/utils/DataHandling.java

@@ -18,8 +18,7 @@ public class DataHandling {
  // Question: Are we assuming all Data Providers know all classes??? Perhaps we can assume yes for now...
  public static Instances @NotNull [] createPartitions(String fullDataSetPath, int noParties)
  throws Exception {
- ConverterUtils.DataSource source = new ConverterUtils.DataSource(fullDataSetPath);
- Instances data = source.getDataSet();
+ Instances data = read_data(fullDataSetPath);
 
  //data.deleteWithMissingClass();
  int numAttributes, noElementsForPartition, noElementsForLastPartition;

src/test/java/TrainingTest.java

@@ -54,7 +54,11 @@ public void train_all() throws Exception {
  while ((line = br.readLine()) != null) {
  String [] values = line.split(",");
  String data_set = values[0];
- train(data_set);
+ String full_data_set_path = new File(data_directory, data_set).toString();
+
+ if (!data_set.endsWith(".model")) {
+ train(full_data_set_path);
+ }
  }
  }
  }