Update heterograph_models.py

AllenInstitute · Oct 30, 2024 · ab1699c · ab1699c
1 parent 9a1bc87
commit ab1699c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/deep_neurographs/machine_learning/heterograph_models.py b/src/deep_neurographs/machine_learning/heterograph_models.py
@@ -255,7 +255,7 @@ def forward(self, x_dict, edge_index_dict, edge_attr_dict):
 
 # --- Utils ---
 def reformat_edge_key(key):
-    if type(key) is str
+    if type(key) is str:
         return tuple([rm_non_alphanumeric(s) for s in key.split(",")])
     else:
         return key