style: Remove DGLHeteroGraph as return type

MuMiN-dataset · Feb 19, 2022 · f419eeb · f419eeb
1 parent 37801c9
commit f419eeb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mumin/dgl.py b/mumin/dgl.py
@@ -288,7 +288,7 @@ def save_dgl_graph(dgl_graph, path: str = 'mumin-small-dgl.bin'):
     save_graphs(path, [dgl_graph])
 
 
-def load_dgl_graph(path: str = 'mumin-small-dgl.bin') -> 'DGLHeteroGraph':
+def load_dgl_graph(path: str = 'mumin-small-dgl.bin'):
     '''Load a MuMiN DGL graph.
 
     Args: