1 files changed, 13 insertions, 6 deletions
diff --git a/lib/param/enum.py b/lib/param/enum.py
index 49e11f6..c68adc9 100644
--- a/lib/param/enum.py
+++ b/lib/param/enum.py
@@ -1,8 +1,15 @@
+"""Defines the enum param type"""
+
 from param import Param
-import sys
 class Enum(Param):
-  def __init__(self, value, enum_values=[], classname=""):
-    super(Enum,self).__init__(classname=classname)
-    if value and value not in enum_values:
-      return self.err_warn("Value %s not in enum values" % str(value))
-    self.value = value
+    """Defines the enum param type
+    Args:
+        value: the value of the param
+        enum_values: an array of possible value types
+        classname: name of the class that the param belongs to
+    """
+    def __init__(self, value, enum_values=[], classname=""):
+        super(Enum, self).__init__(classname=classname)
+        if value and value not in enum_values:
+            self.err_warn("Value %s not in enum values" % str(value))
+        self.value = value