Parcourir la source

update features choice label

Jérôme BUISINE il y a 3 ans
Parent
commit
1ba24e9c6d
1 fichiers modifiés avec 2 ajouts et 2 suppressions
  1. 2 2
      generate/generate_dataset_sequence_file.py

+ 2 - 2
generate/generate_dataset_sequence_file.py

@@ -233,8 +233,8 @@ def main():
 
     for id, feature in enumerate(p_features):
 
-        if feature not in features_choices:
-            raise ValueError("Unknown metric, please select a correct metric : ", features_choices)
+        if feature not in cfg.features_choices_labels:
+            raise ValueError("Unknown metric, please select a correct metric : ", cfg.features_choices_labels)
 
         transformations.append(Transformation(feature, p_params[id], p_size))