Subclass str in enums

Author: TylerYep

torchinfo/enums.py

@@ -4,15 +4,15 @@
 
 
 @unique
-class Mode(Enum):
+class Mode(str, Enum):
     """Enum containing all model modes."""
 
     TRAIN = "train"
     EVAL = "eval"
 
 
 @unique
-class RowSettings(Enum):
+class RowSettings(str, Enum):
     """Enum containing all available row settings."""
 
     DEPTH = "depth"
@@ -21,7 +21,7 @@ class RowSettings(Enum):
 
 
 @unique
-class ColumnSettings(Enum):
+class ColumnSettings(str, Enum):
     """Enum containing all available column settings."""
 
     KERNEL_SIZE = "kernel_size"