sdf export fixed

1 files changed, 9 insertions, 8 deletions

diff --git a/lib/dataset.rb b/lib/dataset.rb
index 49208ce..7037679 100644
--- a/lib/dataset.rb
+++ b/lib/dataset.rb
@@ -432,19 +432,20 @@ module OpenTox
     # Convert dataset to SDF format
     # @return [String] SDF string
     def to_sdf
+
+      export_features = merged_features
+      export_features = transformed_bioactivity_features if export_features.empty? 
+      export_features = bioactivity_features if export_features.empty? 
+      export_feature = export_features.first
+
       sdf = ""
       compounds.each do |compound|
         sdf_lines = compound.sdf.sub(/\$\$\$\$\n/,"").split("\n")
         sdf_lines[0] = compound.smiles
         sdf += sdf_lines.join("\n")
-        bioactivity_features.each do |f|
-          v = values(compound,f)
-          unless v.empty?
-            sdf += "\n> <#{f.name}>\n"
-            sdf += v.uniq.join ","
-            sdf += "\n"
-          end
-        end
+        sdf += "\n> <#{export_feature.name}>\n"
+        sdf += values(compound,export_feature).uniq.join ","
+        sdf += "\n"
         sdf += "\n$$$$\n"
       end
       sdf