added return type

MrCurtis · Sep 12, 2023 · d094d48 · d094d48
1 parent 4930cf4
commit d094d48
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/graph.py b/graph.py
@@ -34,7 +34,7 @@ def __str__(self):
         return self.str()
 
 
-def get_graph(engine, table, primary_key):
+def get_graph(engine, table, primary_key) -> Graph:
     """Construct the graph for a specified data-point
 
     Args: