aboutsummaryrefslogtreecommitdiffstats
path: root/utils/enums.py
diff options
context:
space:
mode:
Diffstat (limited to 'utils/enums.py')
-rw-r--r--utils/enums.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/utils/enums.py b/utils/enums.py
index 2450ff7..22f91b8 100644
--- a/utils/enums.py
+++ b/utils/enums.py
@@ -10,6 +10,10 @@ Define Enums for the BSM flavour ratio analysis
from enum import Enum
+def str_enum(x):
+ return '{0}'.format(str(x).split('.')[-1])
+
+
class DataType(Enum):
REAL = 1
ASIMOV = 2