diff --git a/option_type.py b/option_type.py index 79eb23f..c972998 100644 --- a/option_type.py +++ b/option_type.py @@ -1,2 +1,5 @@ -CALL = 'C' -PUT = 'P' \ No newline at end of file +from enum import Enum + +class OptionType(Enum): + CALL = 'CALL' + PUT = 'PUT' \ No newline at end of file