From 65b69d4c35890a7a2d2992108f0cf4eb5202dd1b Mon Sep 17 00:00:00 2001
From: Christoph Helma <helma@in-silico.ch>
Date: Wed, 1 Jun 2016 10:37:00 +0200
Subject: validation tests fixed

---
 lib/model.rb | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

(limited to 'lib/model.rb')

diff --git a/lib/model.rb b/lib/model.rb
index 988cac9..81f9629 100644
--- a/lib/model.rb
+++ b/lib/model.rb
@@ -33,7 +33,6 @@ module OpenTox
 
         #send(feature_selection_algorithm.to_sym) if feature_selection_algorithm
         save
-        self
       end
 
       def correlation_filter
@@ -203,7 +202,7 @@ module OpenTox
         }.each do |key,value|
           model.neighbor_algorithm_parameters[key] ||= value
         end
-        model.neighbor_algorithm_parameters[:type] = "MP2D" if training_dataset.substances.first.is_a? Compound
+        model.neighbor_algorithm_parameters[:type] ||= "MP2D" if training_dataset.substances.first.is_a? Compound
         model.save
         model
       end
-- 
cgit v1.2.3