Merge pull request #213 from fedorov/add-feature-class-check

ENH: check if feature class name is valid
AIM-Harvard · Mar 6, 2017 · 22baea7 · 22baea7
2 parents 79a84e9 + 0db5834
commit 22baea7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/radiomics/featureextractor.py b/radiomics/featureextractor.py
@@ -283,6 +283,9 @@ def enableFeatureClassByName(self, featureClass, enabled=True):
     """
     Enable or disable all features in given class.
     """
+    if featureClass not in self.getFeatureClassNames():
+      print("Feature class "+featureClass+" is not recognized")
+      return
     if enabled:
       self.enabledFeatures[featureClass] = []
     elif featureClass in self.enabledFeatures: