new_type

ntua-unit-of-control-and-informatics · Oct 30, 2024 · f925bac · f925bac
1 parent 230e544
commit f925bac
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/helpers/json_to_predreq.py b/src/helpers/json_to_predreq.py
@@ -36,6 +36,6 @@ def decode(request):
         task=request.model.task.lower(),
         featurizer=featurizers,
     )
-    if request.model["selectedFeatures"] != []:
-        dataset.select_features(SelectionList=request.model["selectedFeatures"])
+    if len(request.model.selected_features) > 0:
+        dataset.select_features(SelectionList=request.model.selected_features)
     return dataset, jaqpot_row_ids