From 1f8c93e8d4681f2a70e17b233280f03398e88fbf Mon Sep 17 00:00:00 2001
From: Sandeep Menon <snm6477@nyu.edu>
Date: Mon, 5 Dec 2022 01:33:41 -0500
Subject: [PATCH] int_to_region_type dict compatible with ME version v0.5.4

---
 models/modules/common.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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):