folds split on unique compounds instead of data entries

1 files changed, 2 insertions, 2 deletions

diff --git a/test/regression.rb b/test/regression.rb
index fa3b7fb..c25ed2b 100644
--- a/test/regression.rb
+++ b/test/regression.rb
@@ -26,7 +26,7 @@ class LazarRegressionTest < MiniTest::Test
     model = Model::LazarRegression.create(training_dataset, :prediction_algorithm => "OpenTox::Algorithm::Regression.local_fingerprint_regression")
     compound = Compound.from_smiles "NC(=O)OCCC"
     prediction = model.predict compound
-    p prediction[:value]
+    p prediction
     refute_nil prediction[:value]
   end
 
@@ -35,7 +35,7 @@ class LazarRegressionTest < MiniTest::Test
     model = Model::LazarRegression.create(training_dataset, :prediction_algorithm => "OpenTox::Algorithm::Regression.local_physchem_regression")
     compound = Compound.from_smiles "NC(=O)OCCC"
     prediction = model.predict compound
-    p prediction[:value]
+    p prediction
     refute_nil prediction[:value]
   end