fix: add missing subject id when loading feature type

1 files changed, 3 insertions, 2 deletions

diff --git a/lazar.rb b/lazar.rb
index 0c91179..a436569 100644
--- a/lazar.rb
+++ b/lazar.rb
@@ -155,9 +155,10 @@ post '/lazar/?' do
     lazar.metadata[OT.dependentVariables] = prediction_feature.uri
     lazar.metadata[OT.trainingDataset] = dataset_uri
 		lazar.metadata[OT.featureDataset] = feature_dataset_uri
-    if training_activities.feature_type.to_s == "classification"
+    case training_activities.feature_type(@subjectid)
+    when "classification"
       lazar.metadata[RDF.type] = [OT.Model, OTA.ClassificationLazySingleTarget]
-    elsif training_activities.feature_type.to_s == "regression"
+    when "regression"
       lazar.metadata[RDF.type] = [OT.Model, OTA.RegressionLazySingleTarget]
     end