Accept integers in fault configuration directly without list

Fraunhofer-AISEC · Aug 11, 2022 · f97ee2e · f97ee2e
1 parent 14cd660
commit f97ee2e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/controller.py b/controller.py
@@ -46,6 +46,8 @@ def build_ranges(fault_range):
     """
     if isinstance(fault_range, dict):
         return build_ranges_dict(fault_range)
+    if type(fault_range) == int:
+        return range(fault_range, fault_range + 1, 1)
     if len(fault_range) == 3:
         return range(fault_range[0], fault_range[1], fault_range[2])
     elif len(fault_range) == 1: