diff --git a/models/modules/common.py b/models/modules/common.py index 6812910..f4647f2 100644 --- a/models/modules/common.py +++ b/models/modules/common.py @@ -58,7 +58,7 @@ def __int__(self): ConvType.SPATIAL_HYPERCUBE_TEMPORAL_HYPERCROSS: ME.RegionType.HYBRID } -int_to_region_type = {m.value: m for m in ME.RegionType} +int_to_region_type = {m: ME.RegionType(m) for m in range(3)} def convert_region_type(region_type):