A&A fixes

2 files changed, 3 insertions, 3 deletions

diff --git a/fminer.rb b/fminer.rb
index ff192ca..972f9ba 100644
--- a/fminer.rb
+++ b/fminer.rb
@@ -70,7 +70,7 @@ post '/fminer/bbrc/?' do
 
   halt 404, "Please submit a dataset_uri." unless params[:dataset_uri] and  !params[:dataset_uri].nil?
   halt 404, "Please submit a prediction_feature." unless params[:prediction_feature] and  !params[:prediction_feature].nil?
-  prediction_feature = OpenTox::Feature.find params[:prediction_feature]
+  prediction_feature = OpenTox::Feature.find params[:prediction_feature], @subjectid
   training_dataset = OpenTox::Dataset.find "#{params[:dataset_uri]}", @subjectid
   halt 404, "No feature #{params[:prediction_feature]} in dataset #{params[:dataset_uri]}" unless training_dataset.features and training_dataset.features.include?(params[:prediction_feature])
 
@@ -90,7 +90,7 @@ post '/fminer/bbrc/?' do
     if prediction_feature.feature_type == "regression"
       @@bbrc.SetRegression(true)
     else
-      @training_classes = training_dataset.feature_classes(prediction_feature.uri)
+      @training_classes = training_dataset.feature_classes(prediction_feature.uri, @subjectid)
     end
 
     feature_dataset = OpenTox::Dataset.new(nil, @subjectid)
diff --git a/lazar.rb b/lazar.rb
index fa65b84..e9fa264 100644
--- a/lazar.rb
+++ b/lazar.rb
@@ -115,7 +115,7 @@ post '/lazar/?' do
       end
     end
       
-    @training_classes = training_activities.feature_classes(prediction_feature.uri) if prediction_feature.feature_type == "classification"
+    @training_classes = training_activities.feature_classes(prediction_feature.uri, @subjectid) if prediction_feature.feature_type == "classification"
     lazar.prediction_algorithm = "Neighbors.local_svm_regression" if  prediction_feature.feature_type == "regression"
 
     training_activities.data_entries.each do |compound,entry|