attr_accessor :training_dataset removed in Lazar models

1 files changed, 2 insertions, 2 deletions

diff --git a/lib/model.rb b/lib/model.rb
index d89051a..4aa8642 100644
--- a/lib/model.rb
+++ b/lib/model.rb
@@ -25,7 +25,7 @@ module OpenTox
       field :neighbor_algorithm_parameters, type: Hash
 
       attr_accessor :prediction_dataset
-      attr_accessor :training_dataset
+      #attr_accessor :training_dataset
 
       # Create a lazar model from a training_dataset and a feature_dataset
       # @param [OpenTox::Dataset] training_dataset
@@ -116,7 +116,7 @@ module OpenTox
         end
 
         def training_dataset
-          return Dataset.find(training_dataset_id)
+          Dataset.find(training_dataset_id)
         end
 
       end